Home
last modified time | relevance | path

Searched refs:thread_dim_limit (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc71 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap()
140 const auto &limit = device_description.thread_dim_limit(); in ThreadDimOk()
Ddevice_description.h74 const ThreadDim &thread_dim_limit() const { return thread_dim_limit_; } in thread_dim_limit() function
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc857 ThreadDim thread_dim_limit; in CreateDeviceDescription() local
858 thread_dim_limit.x = prop.maxThreadsDim[0]; in CreateDeviceDescription()
859 thread_dim_limit.y = prop.maxThreadsDim[1]; in CreateDeviceDescription()
860 thread_dim_limit.z = prop.maxThreadsDim[2]; in CreateDeviceDescription()
861 builder.set_thread_dim_limit(thread_dim_limit); in CreateDeviceDescription()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc1005 ThreadDim thread_dim_limit; in CreateDeviceDescription() local
1006 thread_dim_limit.x = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1009 thread_dim_limit.y = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1012 thread_dim_limit.z = GpuDriver::GetDeviceAttribute( in CreateDeviceDescription()
1015 builder.set_thread_dim_limit(thread_dim_limit); in CreateDeviceDescription()