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.cc70 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap()
136 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.cc904 ThreadDim thread_dim_limit; in PopulateDeviceDescription() local
905 thread_dim_limit.x = prop.maxThreadsDim[0]; in PopulateDeviceDescription()
906 thread_dim_limit.y = prop.maxThreadsDim[1]; in PopulateDeviceDescription()
907 thread_dim_limit.z = prop.maxThreadsDim[2]; in PopulateDeviceDescription()
908 builder.set_thread_dim_limit(thread_dim_limit); in PopulateDeviceDescription()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc1059 ThreadDim thread_dim_limit; in PopulateDeviceDescription() local
1060 thread_dim_limit.x = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1063 thread_dim_limit.y = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1066 thread_dim_limit.z = GpuDriver::GetDeviceAttribute( in PopulateDeviceDescription()
1069 builder.set_thread_dim_limit(thread_dim_limit); in PopulateDeviceDescription()