Home
last modified time | relevance | path

Searched refs:CreateReshape (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover_test.cc48 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)); in TEST_F()
50 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1)); in TEST_F()
89 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, rng0));
117 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)); in TEST_F()
119 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1)); in TEST_F()
144 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)); in TEST_F()
146 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1)); in TEST_F()
194 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1));
199 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param2));
235 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0));
[all …]
Ddot_decomposer.cc98 HloInstruction::CreateReshape(lhs_shape_r3, lhs)); in DecomposeBatchDot()
100 HloInstruction::CreateReshape(rhs_shape_r3, rhs)); in DecomposeBatchDot()
110 HloInstruction::CreateReshape(lhs_slice_shape_r2, lhs_slice_r3)); in DecomposeBatchDot()
117 HloInstruction::CreateReshape(rhs_slice_shape_r2, rhs_slice_r3)); in DecomposeBatchDot()
145 HloInstruction::CreateReshape(dot_shape_r3, dot_r2)); in DecomposeBatchDot()
155 HloInstruction::CreateReshape(dot_shape, concat)); in DecomposeBatchDot()
216 computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
261 computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
282 original_dot, computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
Dconvolution_group_converter.cc361 add(HloInstruction::CreateReshape(reshaped_filter_shape, filter)); in HandleConvolution()
414 HloInstruction::CreateReshape(reshaped_activation_shape, activation)); in HandleConvolution()
421 add(HloInstruction::CreateReshape(reshaped_filter_shape, filter)); in HandleConvolution()
450 auto reshaped_convolution = HloInstruction::CreateReshape( in HandleConvolution()
Dbfloat16_propagation_test.cc579 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
584 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
637 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
642 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
712 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
717 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
808 builder_cond0.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
813 builder_cond0.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
837 builder_cond1.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
842 builder_cond1.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
Dreshape_mover.cc150 HloInstruction::CreateReshape(new_shape, operand)); in UpdateOperand()
244 HloInstruction::CreateReshape(instruction->shape(), new_elementwise); in PerformSinkReshapeOrTranspose()
Ddynamic_index_splitter.cc78 auto bitcast = parent->AddInstruction(HloInstruction::CreateReshape( in Run()
Dalgebraic_simplifier_test.cc1505 HloInstruction::CreateReshape(ShapeUtil::MakeShape(F32, {6}), op)); in TEST_F()
1508 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
1575 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
1603 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
1946 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
1984 .AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2011 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2018 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2025 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2069 HloInstruction::CreateReshape(ShapeUtil::MakeShape(F32, {4}), add)); in TEST_F()
[all …]
Dhlo_creation_utils.cc111 HloInstruction::CreateReshape(result_shape, operand)); in MakeReshapeHlo()
134 computation->AddInstruction(HloInstruction::CreateReshape( in MakeDynamicSliceHlo()
162 computation->AddInstruction(HloInstruction::CreateReshape( in MakeDynamicUpdateSliceHlo()
Dinstruction_fusion_test.cc118 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {1, 1}), param0)); in TEST_F()
134 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {1, 1}), param0)); in TEST_F()
Dhlo_rematerialization_test.cc64 HloInstruction::CreateReshape(scalar_shape_, param)); in MakeRematerializableComputation()
109 HloInstruction::CreateReshape(scalar_shape_, param)); in MakeRematerializableWhileComputation()
Dalgebraic_simplifier.cc268 return computation_->AddInstruction(HloInstruction::CreateReshape( in StripDim()
277 return computation_->AddInstruction(HloInstruction::CreateReshape( in Flatten()
1181 HloInstruction::CreateReshape(dot->shape(), hlo)); in HandleDotStrengthReduction()
2046 broadcast, HloInstruction::CreateReshape(broadcast->shape(), operand)); in HandleBroadcast()
2615 reshape, HloInstruction::CreateReshape(reshape->shape(), in HandleReshape()
2706 slice, HloInstruction::CreateReshape(slice->shape(), in TrySimplifyScalarSlice()
2814 slice, HloInstruction::CreateReshape(slice->shape(), new_slice))); in TryToReorderSliceAndReshape()
2952 computation_->AddInstruction(HloInstruction::CreateReshape( in HandleReduce()
2959 reduce, HloInstruction::CreateReshape(reduce_result_shape, arg)); in HandleReduce()
3148 HloInstruction::CreateReshape(reduce_window->shape(), reduce)); in HandleReduceWindow()
[all …]
Ddynamic_dimension_inference_test.cc386 HloInstruction::CreateReshape(output_shape, a_param)); in TEST_F()
420 builder.AddInstruction(HloInstruction::CreateReshape(output_shape, a_param)); in TEST_F()
Dlayout_assignment_test.cc359 builder.AddInstruction(HloInstruction::CreateReshape(bshape, log)); in TEST_F()
556 HloInstruction::CreateReshape(cshape, concatenate)); in TEST_F()
644 builder.AddInstruction(HloInstruction::CreateReshape(input_shape, param)); in TEST_F()
Dwhile_loop_simplifier.cc838 lhs = add_new_instr(HloInstruction::CreateReshape(shape, lhs)); in TryMergeInductionVariables()
841 rhs = add_new_instr(HloInstruction::CreateReshape(shape, rhs)); in TryMergeInductionVariables()
Dpattern_matcher_test.cc900 SetName("r", HloInstruction::CreateReshape( in TEST()
906 SetName("r1", HloInstruction::CreateReshape( in TEST()
Dhlo_instruction.cc1149 HloInstruction* reshaped_operand = adder(HloInstruction::CreateReshape( in CreateBroadcastSequence()
1174 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReshape( in CreateReshape() function in xla::HloInstruction
1462 clone = CreateReshape(shape, new_operands[0]); in CloneWithNewOperands()
Dhlo_instruction.h687 static std::unique_ptr<HloInstruction> CreateReshape(const Shape& shape,
Dheap_simulator_test.cc144 cond_builder.AddInstruction(HloInstruction::CreateReshape(r0f32, slice)); in TEST_F()
Dhlo_parser.cc893 HloInstruction::CreateReshape(shape, operands[0])); in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc336 auto reshape = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F()
351 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F()
367 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {3, 2}), const0)); in XLA_TEST_F()
382 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), const0)); in XLA_TEST_F()
396 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F()
412 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), const0)); in XLA_TEST_F()
427 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {3, 3}), const0)); in XLA_TEST_F()
564 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {2, 2}), const0)); in XLA_TEST_F()
Dmultioutput_fusion_test.cc149 builder.AddInstruction(HloInstruction::CreateReshape( in RunTest1D()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc117 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
320 builder.AddInstruction(HloInstruction::CreateReshape(result_shape, exp1)); in TEST_F()
348 HloInstruction::CreateReshape(reshape_shape, broadcast2)); in TEST_F()
431 HloInstruction::CreateReshape(result_shape, param0)); in TEST_F()
597 auto idx_choice = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc81 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), exp1)); in TEST_F()
117 auto reshape2 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
Dcudnn_conv_rewriter.cc487 rhs = c->AddInstruction(HloInstruction::CreateReshape(new_shape, rhs)); in MatchBackwardInput()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc171 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()