/external/tensorflow/tensorflow/compiler/xla/service/ |
D | reshape_mover_test.cc | 48 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 …]
|
D | convolution_4d_expander.cc | 159 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()
|
D | dot_decomposer.cc | 92 computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot() 139 computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot() 165 original_dot, computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
|
D | hlo_rematerialization_test_utils.h | 63 HloInstruction::CreateReshape(scalar_shape_, param)); 108 HloInstruction::CreateReshape(scalar_shape_, param));
|
D | all_to_all_decomposer.cc | 83 instruction->parent()->AddInstruction(HloInstruction::CreateReshape( in ExpandInstruction() 90 HloInstruction::CreateReshape(instruction->shape(), all_to_all)); in ExpandInstruction()
|
D | bfloat16_propagation_test.cc | 609 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()
|
D | convolution_group_converter.cc | 461 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()
|
D | reshape_mover.cc | 151 HloInstruction::CreateReshape(new_shape, operand)); in UpdateOperand() 245 HloInstruction::CreateReshape(instruction->shape(), new_elementwise); in PerformSinkReshapeOrTranspose()
|
D | dynamic_index_splitter.cc | 78 auto bitcast = parent->AddInstruction(HloInstruction::CreateReshape( in Run()
|
D | algebraic_simplifier_test.cc | 1869 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 …]
|
D | all_gather_decomposer.cc | 98 HloInstruction::CreateReshape(ShapeUtil::MakeShape(U32, {}), shard_id)); in DecomposeAllGather()
|
D | hlo_creation_utils.cc | 125 HloInstruction::CreateReshape(result_shape, operand)); in MakeReshapeHlo() 163 computation->AddInstruction(HloInstruction::CreateReshape( in MakeDynamicSliceHlo() 191 computation->AddInstruction(HloInstruction::CreateReshape( in MakeDynamicUpdateSliceHlo()
|
D | dynamic_dimension_inference_test.cc | 575 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()
|
D | algebraic_simplifier.cc | 1792 ? 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/ |
D | reshape.h | 26 GPUOperation CreateReshape(const OperationDef& definition);
|
D | reshape.cc | 87 GPUOperation CreateReshape(const OperationDef& definition) { in CreateReshape() function
|
D | reshape_test_util.cc | 43 GPUOperation operation = CreateReshape(op_def); in ReshapeTest()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | cpu_gpu_fusion_test.cc | 335 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/ |
D | gpu_conv_rewriter.cc | 91 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()
|
D | instruction_fusion_test.cc | 82 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/ |
D | canonicalize_all_gather_for_cse.cc | 88 HloInstruction::CreateReshape(ag->shape(), new_ag)); in RunOnComputation()
|
D | fft_handler.cc | 119 auto reshape_indices_iota = b->AddInstruction(HloInstruction::CreateReshape( in ShuffleWithinEachPartitionUsingOneHot() 130 b->AddInstruction(HloInstruction::CreateReshape( in ShuffleWithinEachPartitionUsingOneHot()
|
D | gather_scatter_handler.cc | 95 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/ |
D | cpu_instruction_fusion_test.cc | 119 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/ |
D | simple_selectors.cc | 126 GPUOperation operation = CreateReshape(op_def); in SelectReshape()
|