/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
D | conv_weights_converter_test_util.cc | 59 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 …]
|
D | max_unpooling.cc | 40 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()
|
D | pooling.cc | 36 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 …]
|
D | split.cc | 33 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()
|
D | elementwise_test_util.cc | 39 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 …]
|
D | depthwise_conv.cc | 82 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()
|
D | padding_test_util.cc | 44 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()
|
D | reshapex4.cc | 30 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()
|
D | reshape.cc | 29 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()
|
D | conv_weights_converter.cc | 128 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()
|
D | resize_test_util.cc | 45 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()
|
D | concat_xy.cc | 57 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()
|
D | pooling_test_util.cc | 47 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()
|
D | transpose.cc | 30 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()
|
D | relu_test_util.cc | 44 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()
|
D | reduce_test_util.cc | 41 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()
|
D | quantize_and_dequantize_test_util.cc | 49 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()
|
D | padding.cc | 30 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()
|
D | space_to_depth_test_util.cc | 40 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()
|
D | softmax_test_util.cc | 42 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()
|
D | conv_powervr_test_util.cc | 50 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()
|
D | lstm.cc | 93 op.AddDstTensor("new_state", definition.dst_tensors[0]); in CreateLSTM() 94 op.AddDstTensor("activation", definition.dst_tensors[1]); in CreateLSTM()
|
D | add_test_util.cc | 42 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()
|
D | prelu_test_util.cc | 47 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()
|
D | concat_test_util.cc | 47 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()
|