Searched defs:VisibleDeviceCount (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ | ||
H A D | platform.cc | 44 int InterpreterPlatform::VisibleDeviceCount() const { return 1; } function in class:perftools::gputools::interpreter::InterpreterPlatform |
/external/tensorflow/tensorflow/stream_executor/host/ | ||
H A D | host_platform.cc | 41 int HostPlatform::VisibleDeviceCount() const { function in class:perftools::gputools::host::HostPlatform |
/external/tensorflow/tensorflow/stream_executor/cuda/ | ||
H A D | cuda_platform.cc | 86 for (int i = 0; i < VisibleDeviceCount(); i++) { 120 for (int i = 0; i < VisibleDeviceCount(); i++) { 135 int CudaPlatform::VisibleDeviceCount() const { function in class:perftools::gputools::cuda::CudaPlatform |
Completed in 172 milliseconds