Searched defs:other_operand (Results 1 – 5 of 5) 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
4042 auto* other_operand = inst->mutable_operand(1); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local