Home
last modified time | relevance | path

Searched defs:GetDeviceCount (Results 1 – 6 of 6) sorted by relevance

/external/armnn/profiling/common/include/
DCounterDirectory.hpp54 …uint16_t GetDeviceCount() const override { return arm::pipe::numeric_cast<uint16_t>(m_Devices.… in GetDeviceCount() function in arm::pipe::CounterDirectory
/external/pytorch/c10/cuda/
DCUDAFunctions.cpp189 cudaError_t GetDeviceCount(int* dev_count) { in GetDeviceCount() function
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test_util.cc160 void GetDeviceCount(const SP_Platform* platform, int* device_count, in GetDeviceCount() function
/external/armnn/src/profiling/test/
DProfilingMocks.hpp578 …uint16_t GetDeviceCount() const override { return arm::pipe::numeric_cast<uint16_t>(m_Devices.… in GetDeviceCount() function in arm::pipe::MockCounterDirectory
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc988 /* static */ int GpuDriver::GetDeviceCount() { in GetDeviceCount() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_driver.cc1272 /* static */ int GpuDriver::GetDeviceCount() { in GetDeviceCount() function in stream_executor::gpu::GpuDriver