Searched refs:num_segments_attr (Results 1 – 2 of 2) sorted by relevance
2741 DenseIntElementsAttr num_segments_attr; in VerifyUnsortedSegmentReduction() local2742 if (matchPattern(op.num_segments(), m_Constant(&num_segments_attr))) { in VerifyUnsortedSegmentReduction()2743 int64_t num_segments = (*num_segments_attr.begin()).getSExtValue(); in VerifyUnsortedSegmentReduction()
4879 DenseIntElementsAttr num_segments_attr; in matchAndRewrite() local4880 if (!matchPattern(op.num_segments(), m_Constant(&num_segments_attr))) in matchAndRewrite()4886 output_shape.push_back((*num_segments_attr.begin()).getSExtValue()); in matchAndRewrite()