Searched refs:Extract1DVector (Results 1 – 1 of 1) sorted by relevance
106 SmallVector<int64_t, 4> Extract1DVector(DenseIntElementsAttr elements) { in Extract1DVector() function1203 Extract1DVector(dim_numbers.lhs_batching_dimensions()); in matchAndRewrite()1205 Extract1DVector(dim_numbers.rhs_batching_dimensions()); in matchAndRewrite()1207 Extract1DVector(dim_numbers.lhs_contracting_dimensions()); in matchAndRewrite()1209 Extract1DVector(dim_numbers.rhs_contracting_dimensions()); in matchAndRewrite()1333 SmallVector<int64_t, 4> reduction_dims = Extract1DVector(op.dimensions()); in matchAndRewrite()