Searched refs:other_operand (Results 1 – 5 of 5) sorted by relevance
77 HloInstruction** other_operand) { in MatchBinaryInstructionOperand() argument81 *other_operand = instruction->mutable_operand(1); in MatchBinaryInstructionOperand()86 *other_operand = instruction->mutable_operand(0); in MatchBinaryInstructionOperand()95 HloInstruction** other_operand) { in MatchBinaryInstructionOperandOpcode() argument100 instruction, matching_operand, other_operand); in MatchBinaryInstructionOperandOpcode()
341 auto other_operand = (next->operands()[0] == prev) in RewriteGraph() local347 if (other_operand->IsCrossModuleAllReduce() && in RewriteGraph()348 other_operand->user_count() == 1) { in RewriteGraph()349 TF_CHECK_OK(other_operand->ReplaceAllUsesWith( in RewriteGraph()350 other_operand->mutable_operand(0))); in RewriteGraph()352 auto shape = other_operand->shape(); in RewriteGraph()359 other_operand, divisor)); in RewriteGraph()360 TF_CHECK_OK(other_operand->ReplaceUseWith(next, division)); in RewriteGraph()
66 HloInstruction** other_operand);73 HloInstruction** other_operand);
788 const int other_operand = 1 - try_operand; in try_emit_mad_for_and_not() local795 ir->operands[other_operand]->accept(this); in try_emit_mad_for_and_not()
1247 const int other_operand = 1 - try_operand; in try_emit_mad_for_and_not() local1254 ir->operands[other_operand]->accept(this); in try_emit_mad_for_and_not()