Searched refs:one_like_x (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 799 Value one_like_x = chlo::getConstantLike(rewriter, loc, 1.0, x); in MaterializeZeta() local
800 Value x_minus_one = rewriter.create<mhlo::SubOp>(loc, x, one_like_x); in MaterializeZeta()
867 loc, rewriter.create<mhlo::CompareOp>(loc, x, one_like_x, kEQ), in MaterializeZeta()
872 loc, rewriter.create<mhlo::CompareOp>(loc, x, one_like_x, kLT), in MaterializeZeta()
|