Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test_utils.h71 auto slice_1 = builder.AddInstruction(HloInstruction::CreateSlice(
80 builder.AddInstruction(HloInstruction::CreateSlice(vec1_shape_, concat_2,
112 HloInstruction::CreateSlice(vec1_shape_, bcast, /*start_indices=*/{0},
120 builder.AddInstruction(HloInstruction::CreateSlice(vec1_shape_, concat,
Dbfloat16_propagation_test.cc569 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
574 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
626 builder_cond.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
631 builder_cond.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
702 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
707 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
798 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
803 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
827 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
832 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
Ddynamic_index_splitter.cc79 auto slice = parent->AddInstruction(HloInstruction::CreateSlice( in Run()
Dall_reduce_contiguous.cc71 HloInstruction::CreateSlice(flat_shape, new_all_reduce, in ReplaceWithContiguousAllReduce()
Ddot_merger.cc206 HloInstruction* new_a = a->AddInstruction(HloInstruction::CreateSlice( in TryMergeSameOperand()
212 HloInstruction* new_b = b->AddInstruction(HloInstruction::CreateSlice( in TryMergeSameOperand()
Dall_to_all_decomposer.cc107 all_to_all->parent()->AddInstruction(HloInstruction::CreateSlice( in ExpandInstruction()
Dwhile_loop_concat_code_motion.cc106 HloInstruction* CreateSlice(HloInstruction* full_data, int64_t element_index, in CreateSlice() function
116 HloInstruction::CreateSlice(shape, full_data, starts, limits, in CreateSlice()
775 group.CreateSlice(concat_output, j, loop->parent()); in RewriteLoopWithConcatGroups()
936 auto slice = operand_group.CreateSlice( in RewriteLoopWithConcatGroups()
Dhlo_creation_utils.cc112 HloInstruction::CreateSlice(slice_shape, operand, start_indices, in MakeSliceHlo()
187 auto slice = computation->AddInstruction(HloInstruction::CreateSlice( in MakeDynamicSliceHlo()
217 auto slice = computation->AddInstruction(HloInstruction::CreateSlice( in MakeDynamicUpdateSliceHlo()
Dhlo_constant_folding_test.cc165 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
Dbuffer_assignment_test.cc1195 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1227 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1264 HloInstruction::CreateSlice(f32vec10_, tuple_element, {0}, {10}, {1})); in TEST_F()
1299 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1331 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1368 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1915 auto root = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
Dhlo_rematerialization_test.cc355 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
512 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_P()
Dalgebraic_simplifier_test.cc2627 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2709 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2760 HloInstruction* slice0 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2766 HloInstruction* slice1 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2771 HloInstruction* slice2 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2776 HloInstruction* slice3 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2781 HloInstruction* slice4 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2786 HloInstruction* slice5 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2791 HloInstruction* slice6 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
2797 HloInstruction* slice7 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_fusion_test.cc76 slice_params.push_back(b.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_reduce_scatter_creator.cc70 rs_input = computation->AddInstruction(HloInstruction::CreateSlice( in Run()
Dcublas_pad_for_gemms.cc103 HloInstruction::CreateSlice(result_shape, new_dot, start_indices, in PadForGemm()
Dcudnn_pad_for_convolutions.cc119 auto* sliced_result = add(HloInstruction::CreateSlice( in PadConv()
Dgpu_conv_padding_legalization.cc391 HloInstruction::CreateSlice(backward_conv_shape, new_backward_conv, in CanonicalizeBackwardInputConvolution()
/external/armnn/src/backends/backendsCommon/
DWorkloadFactoryBase.hpp235 std::unique_ptr<IWorkload> CreateSlice(const SliceQueueDescriptor& /*descriptor*/, in CreateSlice() function in armnn::WorkloadFactoryBase
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc775 HloInstruction::CreateSlice(halo_shape, hlo, halo_start_indices, in ExchangeHalo()
798 concat_pieces.push_back(b->AddInstruction(HloInstruction::CreateSlice( in ExchangeHalo()
836 HloInstruction::CreateSlice(halo_shape, hlo, halo_start_indices, in ExchangeHalo()
961 HloInstruction::CreateSlice(slice_shape, concat, start_indices, in ExchangeHaloAndGetValidData()
1233 return builder->AddInstruction(HloInstruction::CreateSlice( in SliceFirstK()
Dfft_handler.cc342 return b->AddInstruction(HloInstruction::CreateSlice( in SliceValidData()
/external/armnn/include/armnn/backends/
DWorkloadFactory.hpp355 virtual std::unique_ptr<IWorkload> CreateSlice(const SliceQueueDescriptor& descriptor,
/external/armnn/src/backends/neon/
DNeonWorkloadFactory.hpp316 std::unique_ptr<IWorkload> CreateSlice(const SliceQueueDescriptor& descriptor,
/external/armnn/src/backends/reference/
DRefWorkloadFactory.hpp340 std::unique_ptr<IWorkload> CreateSlice(const SliceQueueDescriptor& descriptor,
/external/armnn/src/backends/cl/
DClWorkloadFactory.hpp319 std::unique_ptr<IWorkload> CreateSlice(const SliceQueueDescriptor& descriptor,
/external/webrtc/rtc_base/
Dcopy_on_write_buffer_unittest.cc310 TEST(CopyOnWriteBufferTest, CreateSlice) { in TEST() argument

12