Searched refs:split_sizes_attr (Results 1 – 2 of 2) sorted by relevance
1475 DenseIntElementsAttr split_sizes_attr; in Verify() local1476 if (!matchPattern(op.size_splits(), m_Constant(&split_sizes_attr))) in Verify()1484 split_sizes_attr.getType().cast<ShapedType>().getNumElements()); in Verify()1486 for (auto dim : llvm::enumerate(split_sizes_attr)) { in Verify()
3013 DenseIntElementsAttr split_sizes_attr; in matchAndRewrite() local3014 if (!matchPattern(op.size_splits(), m_Constant(&split_sizes_attr))) in matchAndRewrite()3023 split_sizes_attr.getType().cast<ShapedType>().getNumElements()); in matchAndRewrite()3024 for (auto dim : llvm::enumerate(split_sizes_attr)) { in matchAndRewrite()