Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_generic.cc1242 int blk_total_size = GetRecommendedBlockSize(apple_info, dst_shape); in GetConvParamsForA7A8() local
1244 if (blk_total_size >= 4 && (dst_slices % 4 == 0 || dst_slices >= 16)) { in GetConvParamsForA7A8()
1246 blk_total_size /= 4; in GetConvParamsForA7A8()
1247 } else if (blk_total_size >= 2 && (dst_slices % 2 == 0 || dst_slices >= 4)) { in GetConvParamsForA7A8()
1249 blk_total_size /= 2; in GetConvParamsForA7A8()
1251 if (blk_total_size >= 4) { in GetConvParamsForA7A8()
1254 blk_total_size /= 4; in GetConvParamsForA7A8()
1255 } else if (blk_total_size >= 2) { in GetConvParamsForA7A8()
1261 blk_total_size /= 2; in GetConvParamsForA7A8()
1376 int blk_total_size = GetRecommendedBlockSize(apple_info, dst_shape); in GetConvParamsForA9AndHigher() local
[all …]