Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_metal.cc709 int blk_total_size = GetRecommendedBlockSize(apple_info, dst_shape); in GetConvParamsForA7A8() local
711 if (blk_total_size >= 4 && (dst_slices % 4 == 0 || dst_slices >= 16)) { in GetConvParamsForA7A8()
713 blk_total_size /= 4; in GetConvParamsForA7A8()
714 } else if (blk_total_size >= 2 && (dst_slices % 2 == 0 || dst_slices >= 4)) { in GetConvParamsForA7A8()
716 blk_total_size /= 2; in GetConvParamsForA7A8()
718 if (blk_total_size >= 4) { in GetConvParamsForA7A8()
721 blk_total_size /= 4; in GetConvParamsForA7A8()
722 } else if (blk_total_size >= 2) { in GetConvParamsForA7A8()
728 blk_total_size /= 2; in GetConvParamsForA7A8()
776 int blk_total_size = GetRecommendedBlockSize(apple_info, dst_shape); in GetConvParamsForA9AndHigher() local
[all …]