Searched refs:split_dim_attr (Results 1 – 2 of 2) sorted by relevance
1418 DenseIntElementsAttr split_dim_attr; in VerifySplitInputAndSplitDim() local1419 if (!matchPattern(split_dim, m_Constant(&split_dim_attr))) return success(); in VerifySplitInputAndSplitDim()1421 int64_t index = (*split_dim_attr.begin()).getSExtValue(); in VerifySplitInputAndSplitDim()
2916 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local2917 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite()2922 int64_t dim_index = (*split_dim_attr.begin()).getSExtValue(); in matchAndRewrite()3008 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local3009 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite()3038 int64_t dim_index = (*split_dim_attr.begin()).getSExtValue(); in matchAndRewrite()