Searched refs:new_rhs (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | transpose_folding.cc | 85 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()
|
D | batch_dot_simplification.cc | 51 TF_ASSIGN_OR_RETURN(HloInstruction * new_rhs, in ElideDegenerateBatchDimensionFromBatchDot() 73 MakeDotHlo(new_lhs, new_rhs, new_dim_numbers, in ElideDegenerateBatchDimensionFromBatchDot()
|
D | algebraic_simplifier.cc | 1328 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/ |
D | opt_structure_splitting.cpp | 269 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()
|
D | lower_tess_level.cpp | 319 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()
|
D | lower_distance.cpp | 460 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()
|
D | ast_to_hir.cpp | 980 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/ |
D | cudnn_conv_pad_for_tensor_cores.cc | 84 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()
|