Searched defs:axis_attr (Results 1 – 6 of 6) sorted by relevance
486 IntegerAttr axis_attr = rewriter.getI64IntegerAttr(axis); in matchAndRewrite() local1498 IntegerAttr axis_attr; in matchAndRewrite() local1520 IntegerAttr axis_attr; in matchAndRewrite() local1729 IntegerAttr axis_attr = rewriter.getI32IntegerAttr(0); in matchAndRewrite() local1961 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite() local
1555 IntegerAttr axis_attr; in matchAndRewrite() local2007 IntegerAttr axis_attr; in matchAndRewrite() local2029 IntegerAttr axis_attr; in matchAndRewrite() local2775 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite() local
2339 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in convertReduceOpCommon() local
1086 DenseIntElementsAttr axis_attr; in matchAndRewrite() local1290 DenseIntElementsAttr axis_attr; in Verify() local2432 DenseIntElementsAttr axis_attr; in Verify() local
3478 auto axis_attr = axis->mutable_attr(); in RewriteGraph() local
5374 DenseIntElementsAttr axis_attr; in matchAndRewrite() local