Searched refs:ReplaceInstructionIfSameShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 339 bool ReplaceInstructionIfSameShape(HloInstruction* old_instruction,
479 bool AlgebraicSimplifierVisitor::ReplaceInstructionIfSameShape( in ReplaceInstructionIfSameShape() function in xla::AlgebraicSimplifierVisitor
494 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(add, lhs)) { in HandleAdd()
499 if (IsAll(lhs, 0) && ReplaceInstructionIfSameShape(add, rhs)) { in HandleAdd()
575 if (IsAll(rhs, 1) && ReplaceInstructionIfSameShape(logical_and, lhs)) { in HandleAnd()
581 if (IsAll(lhs, 1) && ReplaceInstructionIfSameShape(logical_and, rhs)) { in HandleAnd()
588 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(logical_and, rhs)) { in HandleAnd()
595 if (IsAll(lhs, 0) && ReplaceInstructionIfSameShape(logical_and, lhs)) { in HandleAnd()
613 ReplaceInstructionIfSameShape(bitcast, bitcast->mutable_operand(0)); in HandleBitcast()
620 ReplaceInstructionIfSameShape(bitcast, bitcast->mutable_operand(0)); in HandleBitcastConvert()
[all …]
|