Home
last modified time | relevance | path

Searched refs:new_rhs (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc85 HloInstruction* new_rhs = dot->mutable_operand(1); in FoldTransposeIntoDot() local
103 CHECK_EQ(new_rhs->opcode(), HloOpcode::kTranspose); in FoldTransposeIntoDot()
106 new_rhs = new_rhs->mutable_operand(0); in FoldTransposeIntoDot()
111 dot->shape(), new_lhs, new_rhs, new_dim_numbers, dot->precision_config()); in FoldTransposeIntoDot()
154 HloInstruction* new_rhs; in FoldTransposeIntoConvolution() local
172 new_rhs = &transpose_operand; in FoldTransposeIntoConvolution()
174 new_rhs = convolution.mutable_operand(kRhsIdx); in FoldTransposeIntoConvolution()
178 convolution.shape(), new_lhs, new_rhs, convolution.feature_group_count(), in FoldTransposeIntoConvolution()
Dbatch_dot_simplification.cc51 TF_ASSIGN_OR_RETURN(HloInstruction * new_rhs, in ElideDegenerateBatchDimensionFromBatchDot()
73 MakeDotHlo(new_lhs, new_rhs, new_dim_numbers, in ElideDegenerateBatchDimensionFromBatchDot()
Dalgebraic_simplifier.cc1328 auto new_rhs = StripDim(rhs, rhs_kept_dim); in HandleDotStrengthReduction() local
1334 multiply(new_lhs, new_rhs), reduce_dim)))); in HandleDotStrengthReduction()
1352 auto new_rhs = broadcast(StripDim(rhs, rhs_kept_dim), lhs->shape(), in HandleDotStrengthReduction() local
1355 dot, reshape_if_necessary(add_reduce_in_f32(multiply(lhs, new_rhs), in HandleDotStrengthReduction()
1671 HloInstruction * new_rhs, in HandleDot()
1684 new_rhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot()
1685 dot->shape(), new_rhs, rhs_broadcast_dims)); in HandleDot()
1689 new_lhs, new_rhs)); in HandleDot()
1708 HloInstruction * new_rhs, in HandleDot()
1735 new_rhs->shape(), new_lhs, lhs_broadcast_dims)); in HandleDot()
[all …]
/external/mesa3d/src/compiler/glsl/
Dopt_structure_splitting.cpp269 ir_dereference *new_lhs, *new_rhs; in visit_leave() local
281 new_rhs = new(mem_ctx) ir_dereference_variable(rhs_entry->components[i]); in visit_leave()
283 new_rhs = new(mem_ctx) in visit_leave()
288 ir->insert_before(new(mem_ctx) ir_assignment(new_lhs, new_rhs)); in visit_leave()
Dlower_tess_level.cpp319 ir_dereference_array *new_rhs = new(ctx) ir_dereference_array( in visit_leave() local
321 this->handle_rvalue((ir_rvalue **) &new_rhs); in visit_leave()
332 ir_assignment *const assign = new(ctx) ir_assignment(new_lhs, new_rhs); in visit_leave()
Dlower_distance.cpp460 ir_dereference_array *new_rhs = new(ctx) ir_dereference_array( in visit_leave() local
462 this->handle_rvalue((ir_rvalue **) &new_rhs); in visit_leave()
473 ir_assignment *const assign = new(ctx) ir_assignment(new_lhs, new_rhs); in visit_leave()
Dast_to_hir.cpp980 ir_rvalue *new_rhs = in do_assignment() local
982 if (new_rhs != NULL) { in do_assignment()
983 rhs = new_rhs; in do_assignment()
4402 ir_rvalue *new_rhs = validate_assignment(state, initializer_loc, in process_initializer() local
4404 if (new_rhs != NULL) { in process_initializer()
4405 rhs = new_rhs; in process_initializer()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores.cc84 auto* new_rhs = PadInstruction(rhs, new_rhs_shape); in PadConv() local
86 CHECK(new_lhs != lhs || new_rhs != rhs) in PadConv()
96 add(conv->CloneWithNewOperands(new_conv_shape, {new_lhs, new_rhs})); in PadConv()