| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 45 for (int i = 0; i < VisibleDeviceCount(); i++) { in InspectNumaNodes()
79 for (int i = 0; i < VisibleDeviceCount(); i++) { in FirstExecutorForBus()
94 int ROCmPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::gpu::ROCmPlatform
|
| D | rocm_platform.h | 63 int VisibleDeviceCount() const override;
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 46 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs()
91 CHECK_LT(gpu_id, gpu_platform->VisibleDeviceCount()); in AvailableGPUMemory()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 83 for (int i = 0; i < VisibleDeviceCount(); i++) { in InspectNumaNodes()
117 for (int i = 0; i < VisibleDeviceCount(); i++) { in FirstExecutorForBus()
132 int CudaPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::gpu::CudaPlatform
|
| D | cuda_platform.h | 63 int VisibleDeviceCount() const override;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_gpu_device.cc | 81 int device_count = platform.ValueOrDie()->VisibleDeviceCount(); in ListPhysicalDevices()
144 for (int i = 0; i < platform.ValueOrDie()->VisibleDeviceCount(); ++i) { in CreateDevices()
|
| D | xla_tpu_device.cc | 316 int device_count = platform->VisibleDeviceCount(); in ListPhysicalDevices()
362 int device_count = platform->VisibleDeviceCount(); in CreateDevices()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_id_utils.h | 55 const int visible_device_count = device_manager->VisibleDeviceCount(); in CheckValidTfDeviceId()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.h | 50 int VisibleDeviceCount() const override;
|
| D | host_platform.cc | 38 int HostPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::host::HostPlatform
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | interpreter_device.cc | 35 if (platform->VisibleDeviceCount() != 1) { in GetInterpreterClient()
|
| D | cpu_device.cc | 35 if (platform->VisibleDeviceCount() <= 0) { in GetCpuClient()
|
| D | tpu_client.cc | 198 if (platform->VisibleDeviceCount() <= 0) { in GetTpuClient()
|
| D | gpu_device.cc | 66 if (platform->VisibleDeviceCount() <= 0) { in GetGpuXlaClient()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.h | 39 int VisibleDeviceCount() const override;
|
| D | platform.cc | 41 int XlaInterpreterPlatform::VisibleDeviceCount() const { return 1; } in VisibleDeviceCount() function in stream_executor::interpreter::XlaInterpreterPlatform
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | platform.h | 124 virtual int VisibleDeviceCount() const = 0;
|
| D | platform.cc | 110 int device_count = VisibleDeviceCount(); in GetPeerAccessMap()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_platform.h | 58 int VisibleDeviceCount() const override;
|
| D | tpu_platform.cc | 77 int TpuPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in tensorflow::tpu::TpuPlatform
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | tensorrt_test.cc | 135 if (CHECK_NOTNULL(GPUMachineManager())->VisibleDeviceCount() <= 0) { in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | platform_util.cc | 168 int device_count = platform->VisibleDeviceCount(); in GetStreamExecutors()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_internal.h | 60 int VisibleDeviceCount() const override { in VisibleDeviceCount() function
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 788 visible_gpu_order->resize(gpu_manager->VisibleDeviceCount()); in ParseVisibleDeviceList()
805 platform_gpu_id >= gpu_manager->VisibleDeviceCount()) { in ParseVisibleDeviceList()
809 gpu_manager->VisibleDeviceCount()); in ParseVisibleDeviceList()
1091 int device_count = gpu_manager->VisibleDeviceCount(); in CacheDeviceIds()
1152 if (gpu_manager->VisibleDeviceCount() <= 0) { in CreateDevices()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 305 int VisibleDeviceCount() const override { return 0; } in TEST() function in tensorflow::__anonde1592d50111::TEST::FakePlatform
|