Searched defs:inputXShape (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | AdditionEndToEndTestImpl.hpp | 18 armnn::INetworkPtr CreateAdditionNetwork(const armnn::TensorShape& inputXShape, in CreateAdditionNetwork()
52 const TensorShape& inputXShape = { 2, 2, 2 }; in AdditionEndToEnd() local
89 const TensorShape& inputXShape = { 2, 2 }; in AdditionEndToEndFloat16() local
|
| H A D | MultiplicationEndToEndTestImpl.hpp | 18 armnn::INetworkPtr CreateMultiplicationNetwork(const armnn::TensorShape& inputXShape, in CreateMultiplicationNetwork()
52 const TensorShape& inputXShape = { 2, 2 }; in MultiplicationEndToEnd() local
77 const TensorShape& inputXShape = { 2, 2 }; in MultiplicationEndToEndFloat16() local
|
| H A D | SubtractionEndToEndTestImpl.hpp | 18 armnn::INetworkPtr CreateSubtractionNetwork(const armnn::TensorShape& inputXShape, in CreateSubtractionNetwork()
52 const TensorShape& inputXShape = { 2, 2 }; in SubtractionEndToEnd() local
77 const TensorShape& inputXShape = { 2, 2 }; in SubtractionEndToEndFloat16() local
|
| H A D | BatchMatMulEndToEndTestImpl.hpp | 18 armnn::INetworkPtr CreateBatchMatMulNetwork(const armnn::TensorShape& inputXShape, in CreateBatchMatMulNetwork()
54 const TensorShape& inputXShape = { 2, 2, 2 }; in BatchMatMulEndToEnd() local
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | BatchMatMulLayer.cpp | 37 TensorShape inputXShape = inputShapes[0]; in InferOutputShapes() local
|
| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | Descriptors.cpp | 469 const TensorShape& inputXShape, in GetAxesNotMul()
|