Home
last modified time | relevance | path

Searched refs:m_InputTensorInfos (Results 1 – 25 of 59) sorted by relevance

123

/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp88 if (workloadInfo.m_InputTensorInfos.size() != expectedSize) in ValidateNumInputs()
92 … to_string(workloadInfo.m_InputTensorInfos.size()) + " have been provided."); in ValidateNumInputs()
523 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate()
557 if (workloadInfo.m_InputTensorInfos.size() != 1) in Validate()
560 workloadInfo.m_InputTensorInfos.size())); in Validate()
564 if (workloadInfo.m_InputTensorInfos.size() != workloadInfo.m_OutputTensorInfos.size()) in Validate()
568 workloadInfo.m_InputTensorInfos.size(), workloadInfo.m_OutputTensorInfos.size())); in Validate()
571 for (std::size_t i = 0; i < workloadInfo.m_InputTensorInfos.size(); ++i) in Validate()
573 if (workloadInfo.m_InputTensorInfos[i].GetNumElements() != in Validate()
636 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate()
[all …]
H A DMakeWorkloadHelper.hpp49 const DataType dataType = !info.m_InputTensorInfos.empty() ? in MakeWorkloadHelper()
50 info.m_InputTensorInfos[0].GetDataType() in MakeWorkloadHelper()
/aosp_15_r20/external/armnn/include/armnn/backends/
H A DWorkload.hpp104 if (!info.m_InputTensorInfos.empty()) in TypedWorkload()
106 expectedInputType = info.m_InputTensorInfos.front().GetDataType(); in TypedWorkload()
112 ARMNN_ASSERT_MSG(std::all_of(std::next(info.m_InputTensorInfos.begin()), in TypedWorkload()
113 info.m_InputTensorInfos.end(), in TypedWorkload()
125 if (!info.m_InputTensorInfos.empty()) in TypedWorkload()
127 expectedInputType = info.m_InputTensorInfos.front().GetDataType(); in TypedWorkload()
156 ARMNN_ASSERT_MSG(std::all_of(info.m_InputTensorInfos.begin(), in MultiTypedWorkload()
157 info.m_InputTensorInfos.end(), in MultiTypedWorkload()
181 if (!info.m_InputTensorInfos.empty()) in FirstInputTypedWorkload()
183 ARMNN_ASSERT_MSG(info.m_InputTensorInfos.front().GetDataType() == DataType, in FirstInputTypedWorkload()
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefWorkloadFactory.cpp35 …auto it = std::find_if(std::begin(info.m_InputTensorInfos), std::end(info.m_InputTensorInfos), che… in IsDataType()
36 if (it != std::end(info.m_InputTensorInfos)) in IsDataType()
159 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
296 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
335 if(IsQuantizedType(info.m_InputTensorInfos[0].GetDataType())) in CreateWorkload()
363 if (info.m_InputTensorInfos.empty() ) in CreateWorkload()
372 … if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateWorkload()
410 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
445 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
458 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
[all …]
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonConvolution2dWorkload.cpp93 …BuildArmComputeTensor(*m_KernelTensor, info.m_InputTensorInfos[1], m_Data.m_Parameters.m_DataLayou… in NeonConvolution2dWorkload()
97 …BuildArmComputeTensor(*m_BiasTensor, info.m_InputTensorInfos[2], m_Data.m_Parameters.m_DataLayout); in NeonConvolution2dWorkload()
131 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in NeonConvolution2dWorkload()
133 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in NeonConvolution2dWorkload()
138 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in NeonConvolution2dWorkload()
150 m_KernelTensorInfo = info.m_InputTensorInfos[1]; in NeonConvolution2dWorkload()
154 m_BiasTensorInfo = info.m_InputTensorInfos[2]; in NeonConvolution2dWorkload()
H A DNeonFullyConnectedWorkload.cpp73 m_WeightsTensorInfo = info.m_InputTensorInfos[1]; in NeonFullyConnectedWorkload()
80 m_BiasesTensorInfo = info.m_InputTensorInfos[2]; in NeonFullyConnectedWorkload()
95 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in NeonFullyConnectedWorkload()
97 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in NeonFullyConnectedWorkload()
100 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in NeonFullyConnectedWorkload()
H A DNeonNormalizationFloatWorkload.cpp80 if (info.m_InputTensorInfos[0].GetShape()[1] != info.m_OutputTensorInfos[0].GetShape()[1] in NeonNormalizationFloatWorkload()
81 || info.m_InputTensorInfos[0].GetShape()[0] != info.m_OutputTensorInfos[0].GetShape()[0] in NeonNormalizationFloatWorkload()
82 || info.m_InputTensorInfos[0].GetShape()[3] != info.m_OutputTensorInfos[0].GetShape()[3] in NeonNormalizationFloatWorkload()
83 || info.m_InputTensorInfos[0].GetShape()[2] != info.m_OutputTensorInfos[0].GetShape()[2]) in NeonNormalizationFloatWorkload()
H A DNeonDepthwiseConvolutionWorkload.cpp142 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in NeonDepthwiseConvolutionWorkload()
144 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in NeonDepthwiseConvolutionWorkload()
147 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in NeonDepthwiseConvolutionWorkload()
H A DNeonBatchMatMulWorkload.cpp138 … = GeneratePermutationVectorOnLastTwoDimensions(info.m_InputTensorInfos[0].GetNumDimensions()); in NeonBatchMatMulWorkload()
139 …const TensorInfo permutedXInfo = armnnUtils::Permuted(info.m_InputTensorInfos[0], permutationXVect… in NeonBatchMatMulWorkload()
152 … = GeneratePermutationVectorOnLastTwoDimensions(info.m_InputTensorInfos[1].GetNumDimensions()); in NeonBatchMatMulWorkload()
153 …const TensorInfo permutedYInfo = armnnUtils::Permuted(info.m_InputTensorInfos[1], permutationYVect… in NeonBatchMatMulWorkload()
H A DNeonMultiplicationWorkload.cpp57 auto convertPolicy = (IsQuantizedType(info.m_InputTensorInfos[0].GetDataType()) || in NeonMultiplicationWorkload()
58 IsQuantizedType(info.m_InputTensorInfos[1].GetDataType())) ? in NeonMultiplicationWorkload()
H A DNeonConvolution3dWorkload.cpp94 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in NeonConvolution3dWorkload()
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DSerializerTestUtils.cpp17 , m_InputTensorInfos(inputInfos) in LayerVerifierBase()
44 CHECK(layer->GetNumInputSlots() == m_InputTensorInfos.size()); in VerifyNameAndConnections()
47 for (unsigned int i = 0; i < m_InputTensorInfos.size(); i++) in VerifyNameAndConnections()
53 CHECK(connectedInfo.GetShape() == m_InputTensorInfos[i].GetShape()); in VerifyNameAndConnections()
54 …CHECK(GetDataTypeName(connectedInfo.GetDataType()) == GetDataTypeName(m_InputTensorInfos[i].GetDat… in VerifyNameAndConnections()
58 … CHECK(connectedInfo.GetQuantizationScales() == m_InputTensorInfos[i].GetQuantizationScales()); in VerifyNameAndConnections()
62 … CHECK(connectedInfo.GetQuantizationScale() == m_InputTensorInfos[i].GetQuantizationScale()); in VerifyNameAndConnections()
64 … CHECK(connectedInfo.GetQuantizationOffset() == m_InputTensorInfos[i].GetQuantizationOffset()); in VerifyNameAndConnections()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefConvolution2dWorkload.cpp18 , m_InputShape(info.m_InputTensorInfos[0].GetShape()) in RefConvolution2dWorkload()
19 , m_FilterShape(info.m_InputTensorInfos[1].GetShape()) in RefConvolution2dWorkload()
23 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in RefConvolution2dWorkload()
H A DRefDepthwiseConvolution2dWorkload.cpp23 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in RefDepthwiseConvolution2dWorkload()
25 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in RefDepthwiseConvolution2dWorkload()
29 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in RefDepthwiseConvolution2dWorkload()
H A DRefConvolution3dWorkload.cpp20 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in RefConvolution3dWorkload()
22 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in RefConvolution3dWorkload()
25 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in RefConvolution3dWorkload()
H A DRefFullyConnectedWorkload.cpp30 , m_InputShape(info.m_InputTensorInfos[0].GetShape()) in RefFullyConnectedWorkload()
31 , m_WeightShape(info.m_InputTensorInfos[1].GetShape()) in RefFullyConnectedWorkload()
33 , m_NumActivations(GetNumActivations(info.m_InputTensorInfos[0])) in RefFullyConnectedWorkload()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClFullyConnectedWorkload.cpp66 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in ClFullyConnectedWorkload()
68 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in ClFullyConnectedWorkload()
71 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in ClFullyConnectedWorkload()
H A DClDepthwiseConvolutionWorkload.cpp97 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in ClDepthwiseConvolutionWorkload()
99 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in ClDepthwiseConvolutionWorkload()
102 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in ClDepthwiseConvolutionWorkload()
H A DClConvolution2dWorkload.cpp139 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in ClConvolution2dWorkload()
141 … detailsInfo.m_WeightsTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[1]); in ClConvolution2dWorkload()
145 … detailsInfo.m_BiasTensorInfo = armnn::Optional<armnn::TensorInfo>(info.m_InputTensorInfos[2]); in ClConvolution2dWorkload()
H A DClBatchMatMulWorkload.cpp144 info.m_InputTensorInfos[0].GetShape(), 3); in ClBatchMatMulWorkload()
148 info.m_InputTensorInfos[1].GetShape(), 3); in ClBatchMatMulWorkload()
156 armnn::TensorInfo strippedInfo = armnnUtils::ReduceDims(info.m_InputTensorInfos[0], 3); in ClBatchMatMulWorkload()
175 armnn::TensorInfo strippedInfo = armnnUtils::ReduceDims(info.m_InputTensorInfos[1], 3); in ClBatchMatMulWorkload()
H A DClMultiplicationWorkload.cpp58 auto convertPolicy = (IsQuantizedType(info.m_InputTensorInfos[0].GetDataType()) || in ClMultiplicationWorkload()
59 IsQuantizedType(info.m_InputTensorInfos[1].GetDataType())) ? in ClMultiplicationWorkload()
H A DClConvolution3dWorkload.cpp98 detailsInfo.m_InputTensorInfos = info.m_InputTensorInfos; in ClConvolution3dWorkload()
/aosp_15_r20/external/armnn/src/backends/tosaReference/
H A DTosaRefWorkloadFactory.cpp36 …auto it = std::find_if(std::begin(info.m_InputTensorInfos), std::end(info.m_InputTensorInfos), che… in IsDataType()
37 if (it != std::end(info.m_InputTensorInfos)) in IsDataType()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DProfilerTests.cpp140 workloadInfo.m_InputTensorInfos.push_back(inputTensorInfo);
160 workloadInfo.m_InputTensorInfos.push_back(inputTensorInfo);
184 workloadInfo.m_InputTensorInfos.push_back(inputTensorInfo);
185 workloadInfo.m_InputTensorInfos.push_back(inputTensorInfo);
/aosp_15_r20/external/armnn/include/armnnTestUtils/
H A DWorkloadTestUtils.hpp29 info.m_InputTensorInfos.push_back(tensorInfo); in AddInputToWorkload()
50 info.m_InputTensorInfos[index] = tensorInfo; in SetWorkloadInput()

123