Searched refs:padded_shape_splits (Results 1 – 1 of 1) sorted by relevance
924 SmallVector<Value, 4> padded_shape_splits( in matchAndRewrite() local944 loc, padded_shape_splits[1 + i], block_shape_splits[i])); in matchAndRewrite()957 SmallVector<Value, 6> reshaped_shape_vals{padded_shape_splits[0]}; in matchAndRewrite()967 reshaped_shape_vals.push_back(padded_shape_splits[i]); in matchAndRewrite()993 auto output_batch = padded_shape_splits[0]; in matchAndRewrite()1003 output_shape_vals.push_back(padded_shape_splits[i]); in matchAndRewrite()