Searched refs:split_dim_attr (Results 1 – 3 of 3) sorted by relevance
326 mlir::ElementsAttr split_dim_attr; in matchAndRewrite() local327 if (!split_dim || !matchPattern(split_dim, m_Constant(&split_dim_attr))) in matchAndRewrite()341 IntegerAttr dim_int = ExtractSingleElementAsInteger(split_dim_attr); in matchAndRewrite()
3321 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local3322 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite()3327 int64_t dim_index = (*split_dim_attr.begin()).getSExtValue(); in matchAndRewrite()3384 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local3385 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite()3390 int64_t dim_index = (*split_dim_attr.begin()).getSExtValue(); in matchAndRewrite()3500 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local3501 if (!matchPattern(op.split_dim(), m_Constant(&split_dim_attr))) in matchAndRewrite()3530 int64_t dim_index = (*split_dim_attr.begin()).getSExtValue(); in matchAndRewrite()
1529 DenseIntElementsAttr split_dim_attr; in VerifySplitInputAndSplitDim() local1530 if (!matchPattern(split_dim, m_Constant(&split_dim_attr))) return success(); in VerifySplitInputAndSplitDim()1532 int64_t index = (*split_dim_attr.begin()).getSExtValue(); in VerifySplitInputAndSplitDim()