Searched refs:other_operand (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_query.cc | 79 HloInstruction** other_operand) { in MatchBinaryInstructionOperand() argument
83 *other_operand = instruction->mutable_operand(1); in MatchBinaryInstructionOperand()
88 *other_operand = instruction->mutable_operand(0); in MatchBinaryInstructionOperand()
97 HloInstruction** other_operand) { in MatchBinaryInstructionOperandOpcode() argument
102 instruction, matching_operand, other_operand); in MatchBinaryInstructionOperandOpcode()
|
| D | hlo_query.h | 67 HloInstruction** other_operand);
74 HloInstruction** other_operand);
|
| D | ar_crs_combiner.cc | 560 auto other_operand = (next->operands()[0] == prev) in RewriteGraph() local
566 if (other_operand->IsCrossModuleAllReduce() && in RewriteGraph()
567 other_operand->user_count() == 1) { in RewriteGraph()
568 TF_CHECK_OK(other_operand->ReplaceAllUsesWith( in RewriteGraph()
569 other_operand->mutable_operand(0))); in RewriteGraph()
571 auto shape = other_operand->shape(); in RewriteGraph()
578 other_operand, divisor)); in RewriteGraph()
579 TF_CHECK_OK(other_operand->ReplaceUseWith(next, division)); in RewriteGraph()
|
| /external/llvm-project/mlir/test/Rewrite/ |
| D | pdl-bytecode.mlir | 458 %other_operand = "test.op"() : () -> i32
460 …"test.op"(%operand, %operand, %operand, %operand, %other_operand) : (i32, i32, i32, i32, i32) -> ()
|
| /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 | 3088 auto other_operand = inst->mutable_operand(1); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3089 auto res = new_operands_set.emplace(other_operand); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3091 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()
|