Searched refs:block_shape_i (Results 1 – 2 of 2) sorted by relevance
883 int64_t block_shape_i = in matchAndRewrite() local886 block_shape_ints.push_back(block_shape_i); in matchAndRewrite()
1353 int64_t block_shape_i = in inferReturnTypes() local1355 return_batch *= block_shape_i; in inferReturnTypes()1356 return_shape[1 + i] = (paddings_sum + input_shape[i + 1]) / block_shape_i; in inferReturnTypes()