Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc84 HloInstruction* new_lhs = dot->mutable_operand(0); in FoldTransposeIntoDot() local
97 CHECK_EQ(new_lhs->opcode(), HloOpcode::kTranspose); in FoldTransposeIntoDot()
100 new_lhs = new_lhs->mutable_operand(0); in FoldTransposeIntoDot()
111 dot->shape(), new_lhs, new_rhs, new_dim_numbers, dot->precision_config()); in FoldTransposeIntoDot()
131 HloInstruction* new_lhs; in FoldTransposeIntoConvolution() local
149 new_lhs = &transpose_operand; in FoldTransposeIntoConvolution()
151 new_lhs = convolution.mutable_operand(kLhsIdx); in FoldTransposeIntoConvolution()
178 convolution.shape(), new_lhs, new_rhs, convolution.feature_group_count(), in FoldTransposeIntoConvolution()
Dbatch_dot_simplification.cc49 TF_ASSIGN_OR_RETURN(HloInstruction * new_lhs, in ElideDegenerateBatchDimensionFromBatchDot()
73 MakeDotHlo(new_lhs, new_rhs, new_dim_numbers, in ElideDegenerateBatchDimensionFromBatchDot()
Dalgebraic_simplifier.cc1327 auto new_lhs = StripDim(lhs, lhs_kept_dim); in HandleDotStrengthReduction() local
1334 multiply(new_lhs, new_rhs), reduce_dim)))); in HandleDotStrengthReduction()
1341 auto new_lhs = broadcast(StripDim(lhs, lhs_kept_dim), rhs->shape(), in HandleDotStrengthReduction() local
1344 dot, reshape_if_necessary(add_reduce_in_f32(multiply(new_lhs, rhs), in HandleDotStrengthReduction()
1658 HloInstruction * new_lhs, in HandleDot()
1667 new_lhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot()
1668 dot->shape(), new_lhs, lhs_broadcast_dims)); in HandleDot()
1689 new_lhs, new_rhs)); in HandleDot()
1701 HloInstruction * new_lhs, in HandleDot()
1734 new_lhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot()
[all …]
/external/mesa3d/src/compiler/glsl/
Dlower_vector_derefs.cpp61 ir_dereference *const new_lhs = (ir_dereference *) deref->array; in visit_enter() local
62 ir->set_lhs(new_lhs); in visit_enter()
69 new_lhs->type, in visit_enter()
70 new_lhs->clone(mem_ctx, NULL), in visit_enter()
73 ir->write_mask = (1 << new_lhs->type->vector_elements) - 1; in visit_enter()
Dlower_tess_level.cpp265 ir_dereference *const new_lhs = (ir_dereference *) expr->operands[0]; in fix_lhs() local
272 new_lhs->clone(mem_ctx, NULL), in fix_lhs()
276 ir->set_lhs(new_lhs); in fix_lhs()
317 ir_dereference_array *new_lhs = new(ctx) ir_dereference_array( in visit_leave() local
332 ir_assignment *const assign = new(ctx) ir_assignment(new_lhs, new_rhs); in visit_leave()
Dopt_structure_splitting.cpp269 ir_dereference *new_lhs, *new_rhs; in visit_leave() local
273 new_lhs = new(mem_ctx) ir_dereference_variable(lhs_entry->components[i]); in visit_leave()
275 new_lhs = new(mem_ctx) in visit_leave()
288 ir->insert_before(new(mem_ctx) ir_assignment(new_lhs, new_rhs)); in visit_leave()
Dlower_distance.cpp412 ir_dereference *const new_lhs = (ir_dereference *) expr->operands[0]; in fix_lhs() local
415 new_lhs->clone(mem_ctx, NULL), in fix_lhs()
418 ir->set_lhs(new_lhs); in fix_lhs()
458 ir_dereference_array *new_lhs = new(ctx) ir_dereference_array( in visit_leave() local
473 ir_assignment *const assign = new(ctx) ir_assignment(new_lhs, new_rhs); in visit_leave()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores.cc83 auto* new_lhs = PadInstruction(lhs, new_lhs_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()