Searched refs:zero_like_a (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 787 Value zero_like_a = chlo::getConstantLike(rewriter, loc, 0.0, a); in MaterializeZeta() local
788 Value neg_power = zero_like_a; in MaterializeZeta()
809 Value horner_sum = zero_like_a; in MaterializeZeta()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1477 HloInstruction* zero_like_a = MakeScalarLike(a, 0); in TryDivideToShift() local
1483 changed_shape, a, zero_like_a, ComparisonDirection::kLt)); in TryDivideToShift()
3701 HloInstruction* zero_like_a = BroadcastZeros( in TryRemainderToAnd() local
3706 ShapeUtil::ChangeElementType(a->shape(), PRED), a, zero_like_a, in TryRemainderToAnd()
|