Searched refs:GetGroupsCountForLinearWH (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_metal.cc | 632 int GetGroupsCountForLinearWH(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCountForLinearWH() function
670 return GetGroupsCountForLinearWH(dst_shape, {32, 1, 1}, {1, 1, 1}); in GetMaximumPossibleWavesCount()
736 int g2 = GetGroupsCountForLinearWH(dst_shape, {32, 1, 1}, params.block_size); in GetConvParamsForA7A8()
810 int g2 = GetGroupsCountForLinearWH(dst_shape, {32, 1, 1}, block_size); in GetConvParamsForA9AndHigher()
886 int g2 = GetGroupsCountForLinearWH(dst_shape, {16, 1, 1}, params.block_size); in GetConvParamsForIntel()
|