Home
last modified time | relevance | path

Searched defs:device_description (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc125 bool ThreadDimOk(const DeviceDescription &device_description, in ThreadDimOk()
149 void CalculateDimensionality(const DeviceDescription &device_description, in CalculateDimensionality()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h53 const se::DeviceDescription& device_description() const { in device_description() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile.h145 string ToString(const DeviceDescription& device_description) const { in ToString()
Dexecutable.cc217 const se::DeviceDescription* device_description = in ExecuteWrapperAfterExecution() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc457 const DeviceDescription &device_description = in VlogOccupancyInfo() local
481 int GpuExecutor::CalculateOccupancy(const DeviceDescription& device_description, in CalculateOccupancy()
498 const DeviceDescription& device_description, in CompareOccupancy()
/external/autotest/client/common_lib/cros/network/
Dinterface.py269 def device_description(self): member in Interface
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc343 int GpuExecutor::CalculateOccupancy(const DeviceDescription& device_description, in CalculateOccupancy()
353 const DeviceDescription& device_description, in CompareOccupancy()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dmlir_compiler.cc178 const auto& device_description = stream_exec->GetDeviceDescription(); in GetGpuVersion() local