Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc59 std::vector<TensorFloat32> dst_tensors; in ConvolutionWeightsConverterTest() local
62 dst_tensors.resize(4); in ConvolutionWeightsConverterTest()
72 dst_tensors[i].shape = BHWC(1, texture_height, texture_width, 4); in ConvolutionWeightsConverterTest()
73 dst_tensors[i].data.resize(4 * texture_width * texture_height); in ConvolutionWeightsConverterTest()
74 memcpy(dst_tensors[i].data.data(), weights_data.data() + sub_size * i, in ConvolutionWeightsConverterTest()
78 dst_tensors.resize(1); in ConvolutionWeightsConverterTest()
79 dst_tensors[0].shape = BHWC(1, 1, 1, flt_count); in ConvolutionWeightsConverterTest()
80 dst_tensors[0].data.resize(flt_count); in ConvolutionWeightsConverterTest()
81 memcpy(dst_tensors[0].data.data(), weights_data.data(), in ConvolutionWeightsConverterTest()
85 std::vector<TensorFloat32> dst_tensors_gpu(dst_tensors.size()); in ConvolutionWeightsConverterTest()
[all …]
Dmax_unpooling.cc40 auto dst_desc = op_def.dst_tensors[0]; in GetMaxUnpoolingKernelCode()
49 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
62 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetMaxUnpoolingKernelCode()
73 std::string src_args = op_def.dst_tensors[0].HasAxis(Axis::DEPTH) in GetMaxUnpoolingKernelCode()
77 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
97 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetMaxUnpoolingKernelCode()
104 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
117 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
Dpooling.cc36 auto dst_desc = op_def.dst_tensors[0]; in GetAveragePoolingKernelCode()
55 if (op_def.dst_tensors[0].HasAxis(axis)) { in GetAveragePoolingKernelCode()
78 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
106 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
133 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
153 auto dst_desc = op_def.dst_tensors[0]; in GetMaxPoolingKernelCode()
159 auto dst_ind_desc = op_def.dst_tensors[1]; in GetMaxPoolingKernelCode()
179 if (op_def.dst_tensors[0].HasAxis(axis)) { in GetMaxPoolingKernelCode()
198 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxPoolingKernelCode()
238 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxPoolingKernelCode()
[all …]
Dsplit.cc33 for (int i = 0; i < definition_.dst_tensors.size(); ++i) { in GetSplitCode()
34 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitCode()
71 for (int i = 0; i < definition_.dst_tensors.size(); ++i) { in GetSplitCode()
85 for (int i = 0; i < definition_.dst_tensors.size(); ++i) { in GetSplitChannelsCode()
86 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitChannelsCode()
115 for (int i = 0; i < definition_.dst_tensors.size(); ++i) { in GetSplitChannelsCode()
Delementwise_test_util.cc39 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AbsTest()
66 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in CosTest()
92 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in CopyTest()
117 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in EluTest()
145 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ExpTest()
173 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in HardSwishTest()
200 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in LogTest()
227 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in NegTest()
253 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in RsqrtTest()
282 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SigmoidTest()
[all …]
Ddepthwise_conv.cc82 auto dst_desc = op_def.dst_tensors[0]; in GenerateDepthwiseConvolutionCode()
93 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
122 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
172 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
179 if (op_def.dst_tensors[0].HasAxis(Axis::HEIGHT)) { in GenerateDepthwiseConvolutionCode()
186 if (op_def.dst_tensors[0].HasAxis(Axis::WIDTH)) { in GenerateDepthwiseConvolutionCode()
216 if (op_def.dst_tensors[0].HasAxis(Axis::WIDTH)) { in GenerateDepthwiseConvolutionCode()
219 if (op_def.dst_tensors[0].HasAxis(Axis::HEIGHT)) { in GenerateDepthwiseConvolutionCode()
222 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
226 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
Dpadding_test_util.cc44 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendWidthTest()
74 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependWidthTest()
104 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendHeightTest()
133 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependHeightTest()
162 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendChannelsTest()
191 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependChannelsTest()
220 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependChannelsX4Test()
249 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingComplexTest()
282 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingReflectWidthTest()
312 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingReflectChannelsTest()
Dreshapex4.cc30 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
44 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
71 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateReshapex4()
Dreshape.cc29 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
48 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
90 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateReshape()
Dconv_weights_converter.cc128 AddDstTensor("dst_tensor0", op_def.dst_tensors[0]); in GetConverterToConvWeightsCode()
129 AddDstTensor("dst_tensor1", op_def.dst_tensors[1]); in GetConverterToConvWeightsCode()
130 AddDstTensor("dst_tensor2", op_def.dst_tensors[2]); in GetConverterToConvWeightsCode()
131 AddDstTensor("dst_tensor3", op_def.dst_tensors[3]); in GetConverterToConvWeightsCode()
141 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetConverterToConvWeightsCode()
Dresize_test_util.cc45 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearAlignedTest()
77 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearNonAlignedTest()
110 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithoutHalfPixelTest()
142 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithHalfPixelTest()
174 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestTest()
206 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestAlignCornersTest()
238 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestHalfPixelCentersTest()
Dconcat_xy.cc57 if (op_def.dst_tensors[0].HasAxis(axis)) { in GetConcatKernelCode()
72 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetConcatKernelCode()
79 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetConcatKernelCode()
121 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateConcatXY()
Dpooling_test_util.cc47 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingTest()
78 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingNonEmptyPaddingTest()
110 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingTest()
142 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingIndicesTest()
143 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingIndicesTest()
Dtranspose.cc30 op_def.dst_tensors[0].HasAxis(Axis::BATCH) ? "B" : "0"; in GetTransposeCode()
33 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetTransposeCode()
103 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateTranspose()
Drelu_test_util.cc44 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUNoClipNoAlphaTest()
73 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUClipTest()
102 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaTest()
131 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaClipTest()
Dreduce_test_util.cc41 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MeanHWTest()
68 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceSumChannelsTest()
95 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceProductChannelsTest()
123 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMaxChannelsTest()
151 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMinChannelsTest()
Dquantize_and_dequantize_test_util.cc49 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits8Test()
86 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
122 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits16Test()
159 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
Dpadding.cc30 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetPaddingCode()
37 op_def.dst_tensors[0].HasAxis(Axis::BATCH) ? "B" : "0"; in GetPaddingCode()
49 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
Dspace_to_depth_test_util.cc40 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
67 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
96 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
127 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x4x4x1BlockSize2Test()
Dsoftmax_test_util.cc42 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxTest()
82 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxBigNumberTest()
112 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1Test()
151 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1BigNumberTest()
Dconv_powervr_test_util.cc50 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVR1x1SimpleWeightsTest()
87 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVR1x1Test()
124 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVRSimpleWeightsTest()
161 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVRTest()
Dlstm.cc93 op.AddDstTensor("new_state", definition.dst_tensors[0]); in CreateLSTM()
94 op.AddDstTensor("activation", definition.dst_tensors[1]); in CreateLSTM()
Dadd_test_util.cc42 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AddTwoEqualTensorsTest()
72 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasMoreChannelsThanSecondTest()
103 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasLessChannelsThanSecond()
Dprelu_test_util.cc47 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUAlphaTest()
79 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUAlphaClipTest()
111 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUHWCAlphaTest()
Dconcat_test_util.cc47 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatWidthTest()
80 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatHeightTest()
115 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
152 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsAlignedx4Test()

1234