Searched defs:VisibleDeviceCount (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/host/ | ||
| D | host_platform.cc | 38 int HostPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::host::HostPlatform |
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ | ||
| D | platform.cc | 42 int XlaInterpreterPlatform::VisibleDeviceCount() const { return 1; } in VisibleDeviceCount() function in stream_executor::interpreter::XlaInterpreterPlatform |
| /external/tensorflow/tensorflow/stream_executor/rocm/ | ||
| D | rocm_platform.cc | 99 int ROCmPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::gpu::ROCmPlatform |
| /external/tensorflow/tensorflow/stream_executor/cuda/ | ||
| D | cuda_platform.cc | 134 int CudaPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::gpu::CudaPlatform |