Searched refs:other_operand (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_query.cc | 84 HloInstruction** other_operand) { in MatchBinaryInstructionOperand() argument
88 *other_operand = instruction->mutable_operand(1); in MatchBinaryInstructionOperand()
93 *other_operand = instruction->mutable_operand(0); in MatchBinaryInstructionOperand()
102 HloInstruction** other_operand) { in MatchBinaryInstructionOperandOpcode() argument
107 instruction, matching_operand, other_operand); in MatchBinaryInstructionOperandOpcode()
|
| D | ar_crs_combiner.cc | 561 auto other_operand = (next->operands()[0] == prev) in RewriteGraph() local
567 if (other_operand->IsCrossModuleAllReduce() && in RewriteGraph()
568 other_operand->user_count() == 1) { in RewriteGraph()
569 TF_CHECK_OK(other_operand->ReplaceAllUsesWith( in RewriteGraph()
570 other_operand->mutable_operand(0))); in RewriteGraph()
572 auto shape = other_operand->shape(); in RewriteGraph()
579 other_operand, divisor)); in RewriteGraph()
580 TF_CHECK_OK(other_operand->ReplaceUseWith(next, division)); in RewriteGraph()
|
| D | hlo_query.h | 70 HloInstruction** other_operand);
77 HloInstruction** other_operand);
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | linalg_grad.cc | 282 Output other_operand, Output input_shape, in EinsumGradWrt() argument
344 Einsum(scope, {output_grad, other_operand}, in EinsumGradWrt()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 220 def _GetGradWrt(output_grad, other_operand, input_shape, input_subs, argument
282 grad_reduced = gen_linalg_ops.einsum([output_grad, other_operand],
|
| /external/mesa3d/src/mesa/program/ |
| D | ir_to_mesa.cpp | 713 const int other_operand = 1 - try_operand; in try_emit_mad_for_and_not() local
720 ir->operands[other_operand]->accept(this); in try_emit_mad_for_and_not()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 4042 auto* other_operand = inst->mutable_operand(1); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
4043 auto res = new_operands_set.emplace(other_operand); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
4045 new_operands.push_back(other_operand); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_glsl_to_tgsi.cpp | 1316 const int other_operand = 1 - try_operand; in try_emit_mad_for_and_not() local
1323 ir->operands[other_operand]->accept(this); in try_emit_mad_for_and_not()
|