Searched defs:other_operand (Results 1 – 7 of 7) sorted by relevance
84 HloInstruction** other_operand) { in MatchBinaryInstructionOperand()102 HloInstruction** other_operand) { in MatchBinaryInstructionOperandOpcode()
561 auto other_operand = (next->operands()[0] == prev) in RewriteGraph() local
282 Output other_operand, Output input_shape, in EinsumGradWrt()
220 def _GetGradWrt(output_grad, other_operand, input_shape, input_subs, argument
713 const int other_operand = 1 - try_operand; in try_emit_mad_for_and_not() local
4042 auto* other_operand = inst->mutable_operand(1); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
1316 const int other_operand = 1 - try_operand; in try_emit_mad_for_and_not() local