Searched refs:GetGroupsCountForLinearWHS (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | conv_generic.cc | 1160 int GetGroupsCountForLinearWHS(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCountForLinearWHS() function
1188 return GetGroupsCountForLinearWHS(dst_shape, {32, 1, 1}, int4(1, 1, 1, 1)); in GetMaximumPossibleWavesCount()
1232 GetGroupsCountForLinearWHS(dst_shape, work_group_size, block_size); in CreateWorkGroupSizeOption()
1414 int g3 = GetGroupsCountForLinearWHS(dst_shape, {32, 1, 1}, params.block_size); in GetConvParamsForA9AndHigher()
|