Searched refs:block_shape_splits (Results 1 – 1 of 1) sorted by relevance
932 SmallVector<Value, 4> block_shape_splits( in matchAndRewrite() local944 loc, padded_shape_splits[1 + i], block_shape_splits[i])); in matchAndRewrite()961 reshaped_shape_vals.push_back(block_shape_splits[i]); in matchAndRewrite()996 rewriter.create<MulOp>(loc, output_batch, block_shape_splits[i]); in matchAndRewrite()