Searched refs:axis_attr (Results 1 – 6 of 6) sorted by relevance
486 IntegerAttr axis_attr = rewriter.getI64IntegerAttr(axis); in matchAndRewrite() local489 tf_argmax_op.input(), axis_attr); in matchAndRewrite()1498 IntegerAttr axis_attr; in matchAndRewrite() local1502 axis_attr = tmpAttr; in matchAndRewrite()1504 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite()1520 IntegerAttr axis_attr; in matchAndRewrite() local1524 axis_attr = tmpAttr; in matchAndRewrite()1526 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite()1729 IntegerAttr axis_attr = rewriter.getI32IntegerAttr(0); in matchAndRewrite() local1735 axis_attr); in matchAndRewrite()[all …]
1555 IntegerAttr axis_attr; in matchAndRewrite() local1561 axis_attr = tmpAttr; in matchAndRewrite()1563 int32_t axis = axis_attr.getInt(); in matchAndRewrite()2007 IntegerAttr axis_attr; in matchAndRewrite() local2011 axis_attr = tmpAttr; in matchAndRewrite()2013 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite()2029 IntegerAttr axis_attr; in matchAndRewrite() local2033 axis_attr = tmpAttr; in matchAndRewrite()2035 int32_t axis_i32 = axis_attr.getInt(); in matchAndRewrite()2775 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite() local[all …]
2339 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in convertReduceOpCommon() local2346 rewriter.create<T>(op->getLoc(), reduce_type, val, axis_attr); in convertReduceOpCommon()
1086 DenseIntElementsAttr axis_attr; in matchAndRewrite() local1087 if (!matchPattern(op.axis(), m_Constant(&axis_attr))) return failure(); in matchAndRewrite()1088 if (axis_attr.getNumElements() != 1) return failure(); in matchAndRewrite()1090 axis_attr.getSplatValue<IntegerAttr>().getValue().getSExtValue(); in matchAndRewrite()1290 DenseIntElementsAttr axis_attr; in Verify() local1291 if (matchPattern(op.axis(), m_Constant(&axis_attr))) { in Verify()1295 assert(axis_attr.getNumElements() == 1 && in Verify()1297 int64_t axis = (*axis_attr.begin()).getSExtValue(); in Verify()2432 DenseIntElementsAttr axis_attr; in Verify() local2433 if (matchPattern(op.axis(), m_Constant(&axis_attr))) { in Verify()[all …]
3478 auto axis_attr = axis->mutable_attr(); in RewriteGraph() local3480 auto* axis_t = (*axis_attr)["value"].mutable_tensor(); in RewriteGraph()
5374 DenseIntElementsAttr axis_attr; in matchAndRewrite() local5375 if (!matchPattern(op.axis(), m_Constant(&axis_attr))) { in matchAndRewrite()5381 int64_t axis = (*axis_attr.begin()).getSExtValue(); in matchAndRewrite()