Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc114 HloInstruction* new_rhs = dot->mutable_operand(1); in FoldTransposeIntoDot() local
131 CHECK_EQ(new_rhs->opcode(), HloOpcode::kTranspose); in FoldTransposeIntoDot()
134 new_rhs->dimensions(new_dim_numbers.rhs_contracting_dimensions(0))); in FoldTransposeIntoDot()
135 new_rhs = new_rhs->mutable_operand(0); in FoldTransposeIntoDot()
140 dot->shape(), new_lhs, new_rhs, new_dim_numbers, dot->precision_config()); in FoldTransposeIntoDot()
183 HloInstruction* new_rhs; in FoldTransposeIntoConvolution() local
201 new_rhs = &transpose_operand; in FoldTransposeIntoConvolution()
203 new_rhs = convolution.mutable_operand(kRhsIdx); in FoldTransposeIntoConvolution()
207 convolution.shape(), new_lhs, new_rhs, convolution.feature_group_count(), in FoldTransposeIntoConvolution()
Dbatch_dot_simplification.cc69 TF_ASSIGN_OR_RETURN(HloInstruction * new_rhs, in ElideDegenerateBatchDimensionFromBatchDot()
92 MakeDotHlo(new_lhs, new_rhs, new_dim_numbers, in ElideDegenerateBatchDimensionFromBatchDot()
Dalgebraic_simplifier.cc1796 HloInstruction* new_rhs = in RemoveDegenerateDimensionFromDot() local
1804 MakeDotHlo(new_lhs, new_rhs, new_dnums, dot->precision_config(), in RemoveDegenerateDimensionFromDot()
2338 HloInstruction * new_rhs, in HandleDot()
2344 if (!ShapeUtil::SameElementType(dot->shape(), new_rhs->shape())) { in HandleDot()
2345 new_rhs = MakeConvertToHlo(new_rhs, dot->shape().element_type()); in HandleDot()
2360 new_rhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot()
2361 dot->shape(), new_rhs, rhs_broadcast_dims)); in HandleDot()
2365 new_lhs, new_rhs)); in HandleDot()
2389 HloInstruction * new_rhs, in HandleDot()
2395 if (!ShapeUtil::SameElementType(dot->shape(), new_rhs->shape())) { 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.cpp320 ir_dereference_array *new_rhs = new(ctx) ir_dereference_array( in visit_leave() local
322 this->handle_rvalue((ir_rvalue **) &new_rhs); in visit_leave()
333 ir_assignment *const assign = new(ctx) ir_assignment(new_lhs, new_rhs); in visit_leave()
Dlower_distance.cpp461 ir_dereference_array *new_rhs = new(ctx) ir_dereference_array( in visit_leave() local
463 this->handle_rvalue((ir_rvalue **) &new_rhs); in visit_leave()
474 ir_assignment *const assign = new(ctx) ir_assignment(new_lhs, new_rhs); in visit_leave()
Dast_to_hir.cpp983 ir_rvalue *new_rhs = in do_assignment() local
985 if (new_rhs != NULL) { in do_assignment()
986 rhs = new_rhs; in do_assignment()
4525 ir_rvalue *new_rhs = validate_assignment(state, initializer_loc, in process_initializer() local
4527 if (new_rhs != NULL) { in process_initializer()
4528 rhs = new_rhs; in process_initializer()