Home
last modified time | relevance | path

Searched defs:VisibleDeviceCount (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_platform.cc37 int HostPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::host::HostPlatform
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dplatform.cc40 int XlaInterpreterPlatform::VisibleDeviceCount() const { return 1; } in VisibleDeviceCount() function in stream_executor::interpreter::XlaInterpreterPlatform
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_platform.cc95 int ROCmPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::gpu::ROCmPlatform
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_platform.cc126 int CudaPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in stream_executor::gpu::CudaPlatform
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor_internal.h61 int VisibleDeviceCount() const override { in VisibleDeviceCount() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_platform.cc78 int TpuPlatform::VisibleDeviceCount() const { in VisibleDeviceCount() function in tensorflow::tpu::TpuPlatform
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_jit_compiled_cpu_function_test.cc291 int VisibleDeviceCount() const override { return 0; } in TEST() function in tensorflow::__anon8ca9750b0111::TEST::FakePlatform