Home
last modified time | relevance | path

Searched refs:PlatformDeviceCount (Results 1 – 9 of 9) sorted by relevance

/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.h145 int PlatformDeviceCount() override;
Dtpu_executor.cc51 int TpuExecutor::PlatformDeviceCount() { in PlatformDeviceCount() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h266 virtual int PlatformDeviceCount() = 0;
Dstream_executor_pimpl.cc248 int StreamExecutor::PlatformDeviceCount() const { in PlatformDeviceCount() function in stream_executor::StreamExecutor
249 return implementation_->PlatformDeviceCount(); in PlatformDeviceCount()
Dstream_executor_pimpl.h345 int PlatformDeviceCount() const;
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h185 int PlatformDeviceCount() override { return GpuDriver::GetDeviceCount(); } in PlatformDeviceCount() function
/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