| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | InferOutputTests.hpp | 20 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 …]
|
| H A D | ShapeInferenceTests.cpp | 31 LayerT* BuildGraph(Graph* graph, const std::vector<TensorShape>& inputShapes, Args &&... args) in BuildGraph()
109 void CreateGraphAndRunTest(const std::vector<TensorShape>& inputShapes, in CreateGraphAndRunTest()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | ConcatEndToEndTestImpl.hpp | 23 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
|
| H A D | ComparisonEndToEndTestImpl.hpp | 23 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
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Concatenation.cpp | 53 const std::vector<Shape>& inputShapes, int32_t axis, T* outputData, in concatenation()
73 const std::vector<Shape>& inputShapes, int32_t axis, in concatenation()
101 std::vector<Shape> inputShapes; in concatenation() local
125 std::vector<Shape> inputShapes; in concatenation() local
|
| D | Densify.cpp | 144 std::vector<Shape> inputShapes; in densify() local
|
| /aosp_15_r20/external/armnn/src/armnnOnnxParser/ |
| H A D | OnnxParser.cpp | 59 … 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
|
| /aosp_15_r20/external/armnn/src/armnnUtils/ |
| H A D | ParserPrototxtFixture.hpp | 122 …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()
|
| /aosp_15_r20/external/armnn/src/armnnConverter/ |
| H A D | ArmnnConverter.cpp | 188 const std::vector<armnn::TensorShape>& inputShapes, in ArmnnConverter()
239 std::map<std::string, armnn::TensorShape> inputShapes; in CreateNetwork() local
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | ConversionUtils.hpp | 483 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()
|
| H A D | Converter.cpp | 709 std::vector<armnn::TensorShape> inputShapes; in ConvertConcatenation() local
|
| /aosp_15_r20/external/armnn/src/backends/tosaCommon/test/ |
| H A D | OneToOneMappingTests.cpp | 78 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
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | StackLayer.cpp | 78 std::vector<TensorShape> inputShapes; in ValidateTensorShapesFromInputs() local
|
| H A D | DepthwiseConvolution2dLayer.cpp | 31 const std::vector<TensorShape>& inputShapes = in SerializeLayerParameters() local
|
| H A D | Convolution2dLayer.cpp | 32 const std::vector<TensorShape>& inputShapes = in SerializeLayerParameters() local
|
| H A D | Convolution3dLayer.cpp | 25 const std::vector<TensorShape>& inputShapes = in SerializeLayerParameters() local
|
| H A D | ConcatLayer.cpp | 308 std::vector<TensorShape> inputShapes; in ValidateTensorShapesFromInputs() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/ |
| H A D | execution_trace_observer.cpp | 189 std::vector<std::string> inputShapes; member
233 const std::string& inputShapes = "[]", in writeJsonNode()
|
| /aosp_15_r20/external/armnn/tests/ |
| H A D | InferenceModel.hpp | 133 std::map<std::string, armnn::TensorShape> inputShapes; in Create() local
298 std::map<std::string, armnn::TensorShape> inputShapes; in Create() local
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | ControlTestHelper.hpp | 212 std::vector<int32_t>& inputShapes, in ConcatenationTest()
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | ConversionUtils.hpp | 545 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
|
| /aosp_15_r20/external/armnn/tests/ExecuteNetwork/ |
| H A D | ArmNNExecutor.cpp | 785 std::map<std::string, armnn::TensorShape> inputShapes; in CreateNetwork() local
|
| /aosp_15_r20/external/armnn/src/armnnTestUtils/ |
| H A D | CreateWorkload.hpp | 2056 std::vector<armnn::TensorShape> inputShapes{{ 2, 3, 2, 5 }, { 2, 3, 2, 5 }}; in CreateConcatWorkloadTest() local
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 865 std::vector<armnn::TensorShape> inputShapes; in OutputTensorInfoFromInputs() local
896 … std::vector<armnn::TensorShape> inputShapes) in OutputTensorInfoFromShapes()
|