Searched refs:GetGroupsCountForLinearWHS (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_metal.cc | 644 int GetGroupsCountForLinearWHS(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCountForLinearWHS() function
672 return GetGroupsCountForLinearWHS(dst_shape, {32, 1, 1}, {1, 1, 1}); in GetMaximumPossibleWavesCount()
737 int g3 = GetGroupsCountForLinearWHS(dst_shape, {32, 1, 1}, params.block_size); in GetConvParamsForA7A8()
811 int g3 = GetGroupsCountForLinearWHS(dst_shape, {32, 1, 1}, block_size); in GetConvParamsForA9AndHigher()
|