Home
last modified time | relevance | path

Searched defs:inputShapes (Results 1 – 21 of 21) sorted by relevance

/external/armnn/src/armnn/test/
DInferOutputTests.hpp20 const std::vector<armnn::TensorShape>& inputShapes, in ArgMinMaxInferOutputShapeImpl()
34 const std::vector<armnn::TensorShape> inputShapes in ArgMinMaxInferOutputShape4dTest() local
53 const std::vector<armnn::TensorShape> inputShapes in ArgMinMaxInferOutputShape3dTest() local
72 const std::vector<armnn::TensorShape> inputShapes in ArgMinMaxInferOutputShape2dTest() local
91 const std::vector<armnn::TensorShape> inputShapes in ArgMinMaxInferOutputShape1dTest() local
149 void PreluInferOutputShapeImpl(const std::vector<armnn::TensorShape>& inputShapes, in PreluInferOutputShapeImpl()
159 const std::vector<armnn::TensorShape> inputShapes in PreluInferOutputShapeSameDimsTest() local
179 const std::vector<armnn::TensorShape> inputShapes in PreluInferOutputShapeInputBiggerTest() local
199 const std::vector<armnn::TensorShape> inputShapes in PreluInferOutputShapeAlphaBiggerTest() local
219 const std::vector<armnn::TensorShape> inputShapes in PreluInferOutputShapeNoMatchTest() local
[all …]
DShapeInferenceTests.cpp31 LayerT* BuildGraph(Graph* graph, const std::vector<TensorShape>& inputShapes, Args &&... args) in BuildGraph()
109 void CreateGraphAndRunTest(const std::vector<TensorShape>& inputShapes, in CreateGraphAndRunTest()
/external/armnn/src/backends/backendsCommon/test/
DConcatEndToEndTestImpl.hpp23 INetworkPtr CreateConcatNetwork(const std::vector<TensorShape>& inputShapes, in CreateConcatNetwork()
61 const std::vector<TensorShape> inputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in ConcatDim0EndToEnd() local
125 const std::vector<TensorShape> inputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in ConcatDim1EndToEnd() local
187 const std::vector<TensorShape> inputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in ConcatDim2EndToEnd() local
248 const std::vector<TensorShape> inputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in ConcatDim3EndToEnd() local
DComparisonEndToEndTestImpl.hpp23 INetworkPtr CreateComparisonNetwork(const std::vector<TensorShape>& inputShapes, in CreateComparisonNetwork()
58 const std::vector<TensorShape> inputShapes{{ 2, 2, 2, 2 }, { 2, 2, 2, 2 }}; in ComparisonSimpleEndToEnd() local
86 const std::vector<TensorShape> inputShapes{{ 1, 2, 2, 3 }, { 1, 1, 1, 3 }}; in ComparisonBroadcastEndToEnd() local
/external/armnn/src/armnnUtils/
DParserPrototxtFixture.hpp122 …ParserPrototxtFixture<TParser>::Setup(const std::map<std::string, armnn::TensorShape>& inputShapes, in Setup()
140 …ParserPrototxtFixture<TParser>::Setup(const std::map<std::string, armnn::TensorShape>& inputShapes) in Setup()
175 const std::map<std::string,armnn::TensorShape>& inputShapes, in SetupOptimizedNetwork()
/external/armnn/src/armnnConverter/
DArmnnConverter.cpp188 const std::vector<armnn::TensorShape>& inputShapes, in ArmnnConverter()
239 std::map<std::string, armnn::TensorShape> inputShapes; in CreateNetwork() local
/external/armnn/src/armnn/layers/
DStackLayer.cpp78 std::vector<TensorShape> inputShapes; in ValidateTensorShapesFromInputs() local
DDepthwiseConvolution2dLayer.cpp31 const std::vector<TensorShape>& inputShapes = in SerializeLayerParameters() local
DConvolution2dLayer.cpp32 const std::vector<TensorShape>& inputShapes = in SerializeLayerParameters() local
DConvolution3dLayer.cpp25 const std::vector<TensorShape>& inputShapes = in SerializeLayerParameters() local
DConcatLayer.cpp308 std::vector<TensorShape> inputShapes; in ValidateTensorShapesFromInputs() local
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp59 … const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromBinary()
76 const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromBinaryFile()
82 … const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromTextFile()
88 … const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromString()
500 std::vector<TensorShape> inputShapes, in ComputeOutputInfo()
738 … const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromTextFile()
754 … const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromBinary()
820 … const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromBinaryFile()
855 … const std::map<std::string, armnn::TensorShape>& inputShapes) in CreateNetworkFromString()
1659 std::vector<TensorShape> inputShapes; in ParseConcat() local
/external/armnn/shim/sl/canonical/
DConversionUtils.hpp483 bool ValidateConcatOutputShape(const std::vector<armnn::TensorShape> & inputShapes, in ValidateConcatOutputShape()
533 std::vector<armnn::TensorShape>& inputShapes, in SwizzleInputs()
556 std::vector<armnn::TensorShape>& inputShapes, in TransposeInputTensors()
DConverter.cpp709 std::vector<armnn::TensorShape> inputShapes; in ConvertConcatenation() local
/external/armnn/src/backends/tosaCommon/test/
DOneToOneMappingTests.cpp78 std::vector<std::vector<int32_t>> inputShapes = { { 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; variable
124 std::vector<std::vector<int32_t>> inputShapes = { { 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; variable
/external/armnn/delegate/test/
DControlTestHelper.hpp212 std::vector<int32_t>& inputShapes, in ConcatenationTest()
/external/armnn/tests/
DInferenceModel.hpp133 std::map<std::string, armnn::TensorShape> inputShapes; in Create() local
298 std::map<std::string, armnn::TensorShape> inputShapes; in Create() local
/external/armnn/tests/ExecuteNetwork/
DArmNNExecutor.cpp785 std::map<std::string, armnn::TensorShape> inputShapes; in CreateNetwork() local
/external/android-nn-driver/
DConversionUtils.hpp545 bool ValidateConcatOutputShape(const std::vector<armnn::TensorShape> & inputShapes, in ValidateConcatOutputShape()
595 std::vector<armnn::TensorShape>& inputShapes, in SwizzleInputs()
618 std::vector<armnn::TensorShape>& inputShapes, in TransposeInputTensors()
2028 std::vector<armnn::TensorShape> inputShapes; in ConvertConcatenation() local
/external/armnn/src/armnnTestUtils/
DCreateWorkload.hpp2056 std::vector<armnn::TensorShape> inputShapes{{ 2, 3, 2, 5 }, { 2, 3, 2, 5 }}; in CreateConcatWorkloadTest() local
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.cpp865 std::vector<armnn::TensorShape> inputShapes; in OutputTensorInfoFromInputs() local
896 … std::vector<armnn::TensorShape> inputShapes) in OutputTensorInfoFromShapes()