Searched refs:n_plus_y (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1591 Value n_plus_y = rewriter.create<AddOp>(loc, iotaN, y); in matchAndRewrite() local
1604 rewriter.create<TF::GreaterEqualOp>(loc, b_false.getType(), n_plus_y, in matchAndRewrite()
1606 rewriter.create<TF::LessOp>(loc, b_false.getType(), n_plus_y, b_rows)); in matchAndRewrite()
1617 mlir::ValueRange({n_plus_y, n_plus_x}), in matchAndRewrite()
|