Searched defs:GetDeviceAttribute (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ | ||
| H A D | cupti_collector.cc | 274 absl::optional<int> GetDeviceAttribute(CUdevice device, in GetDeviceAttribute() function in tensorflow::profiler::__anone45c54f90111::PerDeviceCollector |
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ | ||
| H A D | rocm_driver.cc | 1238 /* static */ port::StatusOr<int> GpuDriver::GetDeviceAttribute( in GetDeviceAttribute() function in stream_executor::gpu::GpuDriver |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ | ||
| H A D | cuda_driver.cc | 1513 /* static */ port::StatusOr<int> GpuDriver::GetDeviceAttribute( in GetDeviceAttribute() function in stream_executor::gpu::GpuDriver |