Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h185 int PlatformDeviceCount() override { return GpuDriver::GetDeviceCount(); } in PlatformDeviceCount() function
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h131 int PlatformDeviceCount() override { return 1; } in PlatformDeviceCount() function
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h162 int PlatformDeviceCount() override { return 1; } in PlatformDeviceCount() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.cc51 int TpuExecutor::PlatformDeviceCount() { in PlatformDeviceCount() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc586 int PlatformDeviceCount() override { return visible_device_count_; } in PlatformDeviceCount() function in stream_executor::__anon00950a620111::CStreamExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc248 int StreamExecutor::PlatformDeviceCount() const { in PlatformDeviceCount() function in stream_executor::StreamExecutor