Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_device.cc57 gpu_info.metal_info.max_work_group_size_z = threadsPerGroup.depth; in CreateGpuInfoFromMetalDevice()
61 gpu_info.metal_info.max_work_group_size_z = 64; in CreateGpuInfoFromMetalDevice()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_info.h357 int max_work_group_size_z; member
407 int max_work_group_size_z; member
Dgpu_info.cc902 return opencl_info.max_work_group_size_z; in GetMaxWorkGroupSizeForZ()
905 return metal_info.max_work_group_size_z; in GetMaxWorkGroupSizeForZ()
923 max_size = std::max(max_size, metal_info.max_work_group_size_z); in GetMaxWorkGroupTotalSize()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_device.cc259 info.opencl_info.max_work_group_size_z = max_work_group_sizes.z; in GpuInfoFromDeviceID()