Home
last modified time | relevance | path

Searched refs:m_OutputTensorInfos (Results 1 – 25 of 38) sorted by relevance

12

/external/armnn/src/backends/backendsCommon/
DWorkloadData.cpp99 if (workloadInfo.m_OutputTensorInfos.size() != expectedSize) in ValidateNumOutputs()
103 … to_string(workloadInfo.m_OutputTensorInfos.size()) + " has been provided."); in ValidateNumOutputs()
524 const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; 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()
574 workloadInfo.m_OutputTensorInfos[i].GetNumElements()) in Validate()
637 const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; in Validate()
662 const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; in Validate()
728 const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; in Validate()
755 const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; in Validate()
[all …]
/external/armnn/include/armnn/backends/
DWorkload.hpp121 if (!info.m_OutputTensorInfos.empty()) in TypedWorkload()
123 expectedOutputType = info.m_OutputTensorInfos.front().GetDataType(); in TypedWorkload()
138 ARMNN_ASSERT_MSG(std::all_of(std::next(info.m_OutputTensorInfos.begin()), in TypedWorkload()
139 info.m_OutputTensorInfos.end(), in TypedWorkload()
163 ARMNN_ASSERT_MSG(std::all_of(info.m_OutputTensorInfos.begin(), in MultiTypedWorkload()
164 info.m_OutputTensorInfos.end(), in MultiTypedWorkload()
187 ARMNN_ASSERT_MSG(std::all_of(info.m_OutputTensorInfos.begin(), in FirstInputTypedWorkload()
188 info.m_OutputTensorInfos.end(), in FirstInputTypedWorkload()
DWorkloadInfo.hpp19 std::vector<TensorInfo> m_OutputTensorInfos; member
/external/armnn/src/armnnSerializer/test/
DSerializerTestUtils.cpp18 , m_OutputTensorInfos(outputInfos) in LayerVerifierBase()
45 CHECK(layer->GetNumOutputSlots() == m_OutputTensorInfos.size()); in VerifyNameAndConnections()
67 for (unsigned int i = 0; i < m_OutputTensorInfos.size(); i++) in VerifyNameAndConnections()
70 CHECK(outputInfo.GetShape() == m_OutputTensorInfos[i].GetShape()); in VerifyNameAndConnections()
71 …CHECK(GetDataTypeName(outputInfo.GetDataType()) == GetDataTypeName(m_OutputTensorInfos[i].GetDataT… in VerifyNameAndConnections()
73 CHECK(outputInfo.GetQuantizationScale() == m_OutputTensorInfos[i].GetQuantizationScale()); in VerifyNameAndConnections()
74 CHECK(outputInfo.GetQuantizationOffset() == m_OutputTensorInfos[i].GetQuantizationOffset()); in VerifyNameAndConnections()
/external/armnn/src/backends/neon/workloads/
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()
DNeonConvolution3dWorkload.cpp95 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in NeonConvolution3dWorkload()
DNeonFullyConnectedWorkload.cpp96 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in NeonFullyConnectedWorkload()
DNeonTransposeConvolution2dWorkload.cpp84 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in NeonTransposeConvolution2dWorkload()
DNeonDepthwiseConvolutionWorkload.cpp143 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in NeonDepthwiseConvolutionWorkload()
DNeonConvolution2dWorkload.cpp132 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in NeonConvolution2dWorkload()
/external/armnn/src/backends/reference/workloads/
DRefConvolution2dWorkload.cpp20 , m_OutputShape(info.m_OutputTensorInfos[0].GetShape()) in RefConvolution2dWorkload()
24 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in RefConvolution2dWorkload()
DRefDepthwiseConvolution2dWorkload.cpp24 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in RefDepthwiseConvolution2dWorkload()
DRefConvolution3dWorkload.cpp21 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in RefConvolution3dWorkload()
DRefFullyConnectedWorkload.cpp32 , m_OutputShape(info.m_OutputTensorInfos[0].GetShape()) in RefFullyConnectedWorkload()
/external/armnn/src/backends/tosaReference/
DTosaRefWorkloadFactory.cpp41 …it = std::find_if(std::begin(info.m_OutputTensorInfos), std::end(info.m_OutputTensorInfos), checkT… in IsDataType()
42 if (it != std::end(info.m_OutputTensorInfos)) in IsDataType()
/external/armnn/src/armnn/
DProfilingDetails.hpp62 PrintInfos(infos.m_InputTensorInfos, "Input", !infos.m_OutputTensorInfos.empty()); in AddDetailsToString()
65 if (!infos.m_OutputTensorInfos.empty()) in AddDetailsToString()
68 PrintInfos(infos.m_OutputTensorInfos, "Output", false); in AddDetailsToString()
/external/armnn/include/armnnTestUtils/
DWorkloadTestUtils.hpp39 info.m_OutputTensorInfos.push_back(tensorInfo); in AddOutputToWorkload()
61 info.m_OutputTensorInfos[index] = tensorInfo; in SetWorkloadOutput()
DMockBackend.hpp134 if (info.m_OutputTensorInfos.empty()) in CreateInput()
139 if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateInput()
/external/armnn/src/backends/cl/workloads/
DClConvolution3dWorkload.cpp99 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in ClConvolution3dWorkload()
DClFullyConnectedWorkload.cpp67 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in ClFullyConnectedWorkload()
DClTransposeConvolution2dWorkload.cpp68 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in ClTransposeConvolution2dWorkload()
DClDepthwiseConvolutionWorkload.cpp98 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in ClDepthwiseConvolutionWorkload()
DClConvolution2dWorkload.cpp140 detailsInfo.m_OutputTensorInfos = info.m_OutputTensorInfos; in ClConvolution2dWorkload()
/external/armnn/src/backends/reference/
DRefWorkloadFactory.cpp40 …it = std::find_if(std::begin(info.m_OutputTensorInfos), std::end(info.m_OutputTensorInfos), checkT… in IsDataType()
41 if (it != std::end(info.m_OutputTensorInfos)) in IsDataType()
367 if (info.m_OutputTensorInfos.empty()) in CreateWorkload()
372 … if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateWorkload()
479 if (info.m_OutputTensorInfos.empty()) in CreateWorkload()
483 … if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateWorkload()
876 if (info.m_OutputTensorInfos.empty()) in CreateInput()
881 if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateInput()
998 if (info.m_OutputTensorInfos.empty()) in CreateOutput()
1002 if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateOutput()
/external/armnn/src/armnn/test/
DProfilerTests.cpp165 workloadInfo.m_OutputTensorInfos.push_back(outputTensorInfo);
191 workloadInfo.m_OutputTensorInfos.push_back(outputTensorInfo);
192 workloadInfo.m_OutputTensorInfos.push_back(outputTensorInfo);

12