Searched refs:DimVector (Results 1 – 1 of 1) sorted by relevance
2366 using DimVector = SmallVector<int64_t, num_dims>; typedef in mlir::mhlo::__anon5ef947ee0111::ConvertAvgPoolGradOp2393 DimVector orig_input_shape(orig_input_shape_values.begin(), in matchAndRewrite()2395 DimVector ksize, strides; in matchAndRewrite()2409 DimVector out_grad_shape( in matchAndRewrite()2411 DimVector low_padding(num_dims, 0); in matchAndRewrite()2412 DimVector high_padding(num_dims, 0); in matchAndRewrite()2413 DimVector interior_padding(num_dims, 0); in matchAndRewrite()2471 auto ones = GetI64ElementsAttr(DimVector(num_dims, 1), &rewriter); in matchAndRewrite()