Home
last modified time | relevance | path

Searched refs:supported_subgroup_sizes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_info.cc358 gpu_info->supported_subgroup_sizes = {32}; in GetGpuInfoFromDeviceDescription()
458 return supported_subgroup_sizes.size() == 1 && in IsWaveSizeEqualTo32()
459 supported_subgroup_sizes[0] == 32; in IsWaveSizeEqualTo32()
483 for (auto subgroup_size : supported_subgroup_sizes) { in SupportsSubGroupWithSize()
Dgpu_info.h367 std::vector<int> supported_subgroup_sizes; member
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_device.cc234 info.supported_subgroup_sizes.push_back(sub_group_sizes[i]); in GpuInfoFromDeviceID()