Home
last modified time | relevance | path

Searched refs:src_tensors (Results 1 – 25 of 85) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc99 gpu_info, shape_0, op_def.src_tensors[0].storage_type, in WinogradFromNode()
100 op_def.src_tensors[0].data_type, op_def.src_tensors[0].layout); in WinogradFromNode()
101 td_0.data_type = op_def.src_tensors[0].data_type; in WinogradFromNode()
102 td_0.layout = op_def.src_tensors[0].layout; in WinogradFromNode()
105 gpu_info, shape_1, op_def.src_tensors[0].storage_type, in WinogradFromNode()
106 op_def.src_tensors[0].data_type, op_def.src_tensors[0].layout); in WinogradFromNode()
107 td_1.data_type = op_def.src_tensors[0].data_type; in WinogradFromNode()
108 td_1.layout = op_def.src_tensors[0].layout; in WinogradFromNode()
115 winograd_up_def.src_tensors.push_back(op_def.src_tensors[0]); in WinogradFromNode()
125 conv_def.src_tensors.push_back(td_0); in WinogradFromNode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise_test_util.cc38 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AbsTest()
65 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in CosTest()
91 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in CopyTest()
116 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in EluTest()
144 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ExpTest()
172 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in HardSwishTest()
199 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in LogTest()
226 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in NegTest()
252 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in RsqrtTest()
281 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SigmoidTest()
[all …]
Dconcat_xy.cc31 std::vector<std::string> tensor_names(op_def.src_tensors.size()); in GetConcatKernelCode()
32 for (int i = 0; i < op_def.src_tensors.size(); ++i) { in GetConcatKernelCode()
50 if (op_def.src_tensors[0].HasAxis(axis) && axis != Axis::BATCH) { in GetConcatKernelCode()
93 for (int i = 0; i < op_def.src_tensors.size(); ++i) { in GetConcatKernelCode()
97 if (op_def.src_tensors[i].HasAxis(Axis::BATCH)) { in GetConcatKernelCode()
117 for (int i = 0; i < definition.src_tensors.size(); ++i) { in CreateConcatXY()
119 op.AddSrcTensor(name, definition.src_tensors[i]); in CreateConcatXY()
Dsplit.cc32 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitCode()
51 if (definition_.src_tensors[0].HasAxis(Axis::BATCH)) { in GetSplitCode()
58 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in GetSplitCode()
84 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitChannelsCode()
90 definition_.src_tensors[0].HasAxis(Axis::BATCH) ? ", B" : ""; in GetSplitChannelsCode()
94 if (definition_.src_tensors[0].HasAxis(Axis::BATCH)) { in GetSplitChannelsCode()
103 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in GetSplitChannelsCode()
Dconcat_test_util.cc45 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatWidthTest()
46 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatWidthTest()
78 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatHeightTest()
79 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatHeightTest()
112 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
113 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
114 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
150 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsAlignedx4Test()
151 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsAlignedx4Test()
Dadd_test_util.cc40 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddTwoEqualTensorsTest()
41 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddTwoEqualTensorsTest()
70 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasMoreChannelsThanSecondTest()
71 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasMoreChannelsThanSecondTest()
101 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasLessChannelsThanSecond()
102 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasLessChannelsThanSecond()
Dconcat_z.cc39 std::vector<std::string> tensor_names(op_def.src_tensors.size()); in GetConcatKernelCode()
40 for (int i = 0; i < op_def.src_tensors.size(); ++i) { in GetConcatKernelCode()
126 for (int i = 0; i < definition.src_tensors.size(); ++i) { in CreateConcatZ()
128 auto src_desc = definition.src_tensors[i]; in CreateConcatZ()
147 definition.src_tensors[0].storage_type != TensorStorageType::BUFFER && in CreateConcatZ()
Dpadding_test_util.cc43 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendWidthTest()
73 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependWidthTest()
103 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendHeightTest()
132 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependHeightTest()
161 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendChannelsTest()
190 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependChannelsTest()
219 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependChannelsX4Test()
248 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingComplexTest()
281 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingReflectWidthTest()
311 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingReflectChannelsTest()
Dconvolution_transposed_3x3_thin.cc43 auto src_desc = op_def.src_tensors[0]; in GenerateConvolutionTransposedCode()
48 if (op_def.src_tensors.size() == 2) { in GenerateConvolutionTransposedCode()
51 desc.element_type = op_def.src_tensors[1].data_type; in GenerateConvolutionTransposedCode()
57 const auto src_tensor_type = op_def.src_tensors[0].storage_type; in GenerateConvolutionTransposedCode()
254 new_def.src_tensors = { in CreateConvolutionTransposed3x3ThinDynamicWeights()
255 definition.src_tensors[0]}; // leaving only src_tensor def, weights defs in CreateConvolutionTransposed3x3ThinDynamicWeights()
259 new_def.src_tensors.push_back( in CreateConvolutionTransposed3x3ThinDynamicWeights()
Dpadding.cc29 op->AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetPaddingCode()
66 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
73 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
101 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
Dresize_test_util.cc44 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearAlignedTest()
76 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearNonAlignedTest()
109 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithoutHalfPixelTest()
141 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithHalfPixelTest()
173 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestTest()
205 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestAlignCornersTest()
237 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestHalfPixelCentersTest()
Dconvolution_transposed.cc130 auto src_desc = op_def.src_tensors[0]; in GenerateConvolutionTransposedCode()
135 if (op_def.src_tensors.size() != 1) { in GenerateConvolutionTransposedCode()
140 desc.element_type = op_def.src_tensors[1].data_type; in GenerateConvolutionTransposedCode()
147 desc.element_type = op_def.src_tensors[1 + i].data_type; in GenerateConvolutionTransposedCode()
154 const auto& src_def = op_def.src_tensors[0]; in GenerateConvolutionTransposedCode()
551 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in BindArguments()
617 new_def.src_tensors = { in CreateConvolutionTransposedDynamicWeights()
618 definition.src_tensors[0]}; // leaving only src_tensor def, weights defs in CreateConvolutionTransposedDynamicWeights()
623 new_def.src_tensors.push_back( in CreateConvolutionTransposedDynamicWeights()
627 new_def.src_tensors.push_back( in CreateConvolutionTransposedDynamicWeights()
[all …]
Dtranspose.cc59 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetTransposeCode()
74 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetTransposeCode()
102 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateTranspose()
Drelu_test_util.cc43 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUNoClipNoAlphaTest()
72 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUClipTest()
101 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaTest()
130 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaClipTest()
Dreduce_test_util.cc40 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MeanHWTest()
67 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceSumChannelsTest()
94 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceProductChannelsTest()
122 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMaxChannelsTest()
150 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMinChannelsTest()
Dquantize_and_dequantize_test_util.cc48 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits8Test()
85 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
121 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits16Test()
158 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
Dpooling.cc30 auto src_desc = op_def.src_tensors[0]; in GetAveragePoolingKernelCode()
58 if (op_def.src_tensors[0].HasAxis(axis)) { in GetAveragePoolingKernelCode()
72 op_def.src_tensors[0].storage_type == TensorStorageType::BUFFER || in GetAveragePoolingKernelCode()
73 op_def.src_tensors[0].storage_type == TensorStorageType::IMAGE_BUFFER; in GetAveragePoolingKernelCode()
148 auto src_desc = op_def.src_tensors[0]; in GetMaxPoolingKernelCode()
182 if (op_def.src_tensors[0].HasAxis(axis)) { in GetMaxPoolingKernelCode()
Dspace_to_depth_test_util.cc39 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
66 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
95 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
126 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x4x4x1BlockSize2Test()
Dadd.cc36 for (int i = 1; i < definition.src_tensors.size(); ++i) { in CreateAdd()
38 auto src_desc = definition.src_tensors[i]; in CreateAdd()
Dconvolution_transposed_4x4.cc74 auto src_desc = op_def.src_tensors[0]; in GenerateConvolutionTransposedCode()
87 if (op_def.src_tensors.size() == 2) { in GenerateConvolutionTransposedCode()
90 desc.element_type = op_def.src_tensors[1].data_type; in GenerateConvolutionTransposedCode()
447 new_def.src_tensors = { in CreateConvolutionTransposed4x4DynamicWeights()
448 definition.src_tensors[0]}; // leaving only src_tensor def, weights defs in CreateConvolutionTransposed4x4DynamicWeights()
452 new_def.src_tensors.push_back( in CreateConvolutionTransposed4x4DynamicWeights()
Dreshapex4.cc56 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
70 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateReshapex4()
Dpooling_test_util.cc46 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingTest()
77 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingNonEmptyPaddingTest()
109 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingTest()
141 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingIndicesTest()
Dreshape.cc64 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
89 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateReshape()
Dsoftmax_test_util.cc41 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxTest()
81 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxBigNumberTest()
111 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1Test()
150 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1BigNumberTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc55 return src_tensors[0].data_type; in GetPrimaryDataType()
58 return src_tensors[0].storage_type; in GetPrimaryStorageType()
62 for (const auto& src : src_tensors) { in IsBatchSupported()
146 definition_.src_tensors.push_back( in AddOperation()
147 operation->definition_.src_tensors[i + 1]); in AddOperation()
189 absl::make_unique<TensorDescriptor>(definition_.src_tensors[0]); in AssembleCode()

1234