Searched refs:tf_pack_op (Results 1 – 2 of 2) sorted by relevance
289 auto tf_pack_op = cast<TF::PackOp>(op); in matchAndRewrite() local291 SmallVector<Value, 4> values(tf_pack_op.values()); in matchAndRewrite()292 auto output_type = tf_pack_op.output().getType(); in matchAndRewrite()293 auto values_count = rewriter.getI32IntegerAttr(tf_pack_op.N()); in matchAndRewrite()295 auto axis = rewriter.getI32IntegerAttr(tf_pack_op.axis()); in matchAndRewrite()
1492 auto tf_pack_op = cast<TF::PackOp>(op); in matchAndRewrite() local1494 SmallVector<Value, 8> inputs(tf_pack_op.values()); in matchAndRewrite()1500 auto tmpAttr = tf_pack_op.axisAttr(); in matchAndRewrite()1507 convertPackOp(rewriter, op, tf_pack_op.getResult(), inputs, axis_i32); in matchAndRewrite()