Home
last modified time | relevance | path

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

12

/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 …]
Dconvolution_4d_expander.cc159 HloInstruction::CreateReshape(new_input_shape, input)); in ExpandInstruction()
161 HloInstruction::CreateReshape(new_kernel_shape, kernel)); in ExpandInstruction()
172 HloInstruction::CreateReshape(instruction->shape(), new_convolution)); in ExpandInstruction()
Ddot_decomposer.cc92 computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
139 computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
165 original_dot, computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
Dhlo_rematerialization_test_utils.h63 HloInstruction::CreateReshape(scalar_shape_, param));
108 HloInstruction::CreateReshape(scalar_shape_, param));
Dall_to_all_decomposer.cc83 instruction->parent()->AddInstruction(HloInstruction::CreateReshape( in ExpandInstruction()
90 HloInstruction::CreateReshape(instruction->shape(), all_to_all)); in ExpandInstruction()
Dbfloat16_propagation_test.cc609 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
614 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
667 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
672 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
742 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
747 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
838 builder_cond0.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
843 builder_cond0.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
867 builder_cond1.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
872 builder_cond1.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
Dconvolution_group_converter.cc461 add(HloInstruction::CreateReshape(reshaped_filter_shape, filter)); in HandleConvolution()
514 computation_->AddInstruction(HloInstruction::CreateReshape( in HandleConvolution()
528 computation_->AddInstruction(HloInstruction::CreateReshape( in HandleConvolution()
573 HloInstruction::CreateReshape(convolution->shape(), new_convolution)); in HandleConvolution()
Dreshape_mover.cc151 HloInstruction::CreateReshape(new_shape, operand)); in UpdateOperand()
245 HloInstruction::CreateReshape(instruction->shape(), new_elementwise); in PerformSinkReshapeOrTranspose()
Ddynamic_index_splitter.cc78 auto bitcast = parent->AddInstruction(HloInstruction::CreateReshape( in Run()
Dalgebraic_simplifier_test.cc1869 HloInstruction::CreateReshape(ShapeUtil::MakeShape(F32, {6}), op)); in TEST_F()
1872 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2028 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2056 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2484 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2522 .AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2549 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2556 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2563 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
2607 HloInstruction::CreateReshape(ShapeUtil::MakeShape(F32, {4}), add)); in TEST_F()
[all …]
Dall_gather_decomposer.cc98 HloInstruction::CreateReshape(ShapeUtil::MakeShape(U32, {}), shard_id)); in DecomposeAllGather()
Dhlo_creation_utils.cc125 HloInstruction::CreateReshape(result_shape, operand)); in MakeReshapeHlo()
163 computation->AddInstruction(HloInstruction::CreateReshape( in MakeDynamicSliceHlo()
191 computation->AddInstruction(HloInstruction::CreateReshape( in MakeDynamicUpdateSliceHlo()
Ddynamic_dimension_inference_test.cc575 HloInstruction::CreateReshape(output_shape, a_param)); in TEST_F()
609 auto* reshape = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
636 HloInstruction::CreateReshape(output_shape, a_param)); in TEST_F()
661 builder.AddInstruction(HloInstruction::CreateReshape(output_shape, a_param)); in TEST_F()
1295 auto dynamic_reshape = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
Dalgebraic_simplifier.cc1792 ? dot->parent()->AddInstruction(HloInstruction::CreateReshape( in RemoveDegenerateDimensionFromDot()
1798 ? dot->parent()->AddInstruction(HloInstruction::CreateReshape( in RemoveDegenerateDimensionFromDot()
1810 dot, HloInstruction::CreateReshape(dot->shape(), new_dot))); in RemoveDegenerateDimensionFromDot()
2209 HloInstruction::CreateReshape(reshape->shape(), lhs)); in OptimizeDotOfReorderContractingDims()
2246 computation_->AddInstruction(HloInstruction::CreateReshape( in OptimizeDotOfReorderContractingDims()
2291 HloInstruction::CreateReshape(constant->shape(), rhs)); in OptimizeDotOfReorderContractingDims()
2541 auto scalar = computation_->AddInstruction(HloInstruction::CreateReshape( in HandleGather()
3057 broadcast, HloInstruction::CreateReshape(broadcast->shape(), operand)); in HandleBroadcast()
3154 operand->parent()->AddInstruction(HloInstruction::CreateReshape( in HandleBroadcast()
3899 reshape, HloInstruction::CreateReshape(reshape->shape(), in HandleReshape()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dreshape.h26 GPUOperation CreateReshape(const OperationDef& definition);
Dreshape.cc87 GPUOperation CreateReshape(const OperationDef& definition) { in CreateReshape() function
Dreshape_test_util.cc43 GPUOperation operation = CreateReshape(op_def); in ReshapeTest()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc335 auto reshape = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F()
350 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F()
366 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {3, 2}), const0)); in XLA_TEST_F()
381 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), const0)); in XLA_TEST_F()
395 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F()
411 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), const0)); in XLA_TEST_F()
426 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {3, 3}), const0)); in XLA_TEST_F()
563 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {2, 2}), const0)); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter.cc91 lhs = add(HloInstruction::CreateReshape( in ConvertBatchGroupedToFeatureGroupedConvolution()
111 lhs = add(HloInstruction::CreateReshape(new_shape, lhs)); in ConvertBatchGroupedToFeatureGroupedConvolution()
578 rhs = c->AddInstruction(HloInstruction::CreateReshape( in MatchBackwardInput()
604 rhs = c->AddInstruction(HloInstruction::CreateReshape(new_shape, rhs)); in MatchBackwardInput()
Dinstruction_fusion_test.cc82 HloInstruction::CreateReshape(ShapeUtil::MakeShape(F32, {}), exp1)); in TEST_F()
118 auto reshape2 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcanonicalize_all_gather_for_cse.cc88 HloInstruction::CreateReshape(ag->shape(), new_ag)); in RunOnComputation()
Dfft_handler.cc119 auto reshape_indices_iota = b->AddInstruction(HloInstruction::CreateReshape( in ShuffleWithinEachPartitionUsingOneHot()
130 b->AddInstruction(HloInstruction::CreateReshape( in ShuffleWithinEachPartitionUsingOneHot()
Dgather_scatter_handler.cc95 min_indices[i] = b->AddInstruction(HloInstruction::CreateReshape( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
98 max_indices[i] = b->AddInstruction(HloInstruction::CreateReshape( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
383 ? b->AddInstruction(HloInstruction::CreateReshape( in PartitionIndexParallelDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc119 builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
322 builder.AddInstruction(HloInstruction::CreateReshape(result_shape, exp1)); in TEST_F()
350 HloInstruction::CreateReshape(reshape_shape, broadcast2)); in TEST_F()
433 HloInstruction::CreateReshape(result_shape, param0)); in TEST_F()
599 auto idx_choice = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.cc126 GPUOperation operation = CreateReshape(op_def); in SelectReshape()

12