Home
last modified time | relevance | path

Searched refs:old_operand (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion.cc64 auto get_new_operand = [&](HloInstruction* old_operand) { in CreateLoopInvariantCopy() argument
65 return old_operand == while_body_param in CreateLoopInvariantCopy()
67 : FindOrDie(*hoisted_instructions, old_operand); in CreateLoopInvariantCopy()
Dhlo_module.cc427 HloInstruction* old_operand = in OutlineExpressionFromComputation() local
430 HloInstruction** operand_slot = &(outlined_instructions[old_operand]); in OutlineExpressionFromComputation()
436 arguments.push_back(old_operand); in OutlineExpressionFromComputation()
438 parameter_count, old_operand->shape(), "p")); in OutlineExpressionFromComputation()
Dhlo_instruction.cc1919 auto old_operand = operand(operand_num); in ReplaceOperandWith() local
1920 TF_RET_CHECK(ShapeUtil::CompatibleIgnoringFpPrecision(old_operand->shape(), in ReplaceOperandWith()
1922 << old_operand->shape() << " is not compatible with " in ReplaceOperandWith()
1931 HloInstruction* old_operand = mutable_operand(operand_num); in ReplaceOperandWithDifferentShape() local
1932 if (old_operand == new_operand) { in ReplaceOperandWithDifferentShape()
1939 << new_operand->name() << ", was " << old_operand->name(); in ReplaceOperandWithDifferentShape()
1941 if (!absl::c_linear_search(operands_, old_operand)) { in ReplaceOperandWithDifferentShape()
1942 old_operand->RemoveUser(this); in ReplaceOperandWithDifferentShape()