Searched defs:inputYShape (Results 1 – 6 of 6) sorted by relevance
| /external/armnn/src/backends/backendsCommon/test/ |
| D | SubtractionEndToEndTestImpl.hpp | 19 const armnn::TensorShape& inputYShape, in CreateSubtractionNetwork()
53 const TensorShape& inputYShape = { 2, 2 }; in SubtractionEndToEnd() local
78 const TensorShape& inputYShape = { 2, 2 }; in SubtractionEndToEndFloat16() local
|
| D | AdditionEndToEndTestImpl.hpp | 19 const armnn::TensorShape& inputYShape, in CreateAdditionNetwork()
53 const TensorShape& inputYShape = { 2, 2, 2 }; in AdditionEndToEnd() local
90 const TensorShape& inputYShape = { 2, 2 }; in AdditionEndToEndFloat16() local
|
| D | MultiplicationEndToEndTestImpl.hpp | 19 const armnn::TensorShape& inputYShape, in CreateMultiplicationNetwork()
53 const TensorShape& inputYShape = { 2, 2 }; in MultiplicationEndToEnd() local
78 const TensorShape& inputYShape = { 2, 2 }; in MultiplicationEndToEndFloat16() local
|
| D | BatchMatMulEndToEndTestImpl.hpp | 19 const armnn::TensorShape& inputYShape, in CreateBatchMatMulNetwork()
55 const TensorShape& inputYShape = { 2, 2, 2 }; in BatchMatMulEndToEnd() local
|
| /external/armnn/src/armnn/layers/ |
| D | BatchMatMulLayer.cpp | 38 TensorShape inputYShape = inputShapes[1]; in InferOutputShapes() local
|
| /external/armnn/src/armnn/ |
| D | Descriptors.cpp | 470 const TensorShape& inputYShape) in GetAxesNotMul()
|