| /external/armnn/src/armnn/test/ |
| D | InferOutputTests.hpp | 25 outputShapes = argMinMaxLayer->InferOutputShapes(inputShapes); in ArgMinMaxInferOutputShapeImpl()
124 CHECK(expectedShape == batchToSpaceLayer->InferOutputShapes(shapes).at(0)); in BatchToSpaceInferOutputShapeTest()
146 CHECK(expectedShape == spaceToDepthLayer->InferOutputShapes(shapes).at(0)); in SpaceToDepthInferOutputShapeTest()
154 outputShapes = preluLayer->InferOutputShapes(inputShapes); in PreluInferOutputShapeImpl()
287 outputShapes = stackLayer->InferOutputShapes(inputShapes); in StackInferOutputShapeImpl()
464 CHECK(expectedOutputShape == convolution2dLayer->InferOutputShapes(shapes).at(0)); in Convolution2dInferOutputShapeTest()
501 CHECK(expectedOutputShape == convolution3dLayer->InferOutputShapes(shapes).at(0)); in Convolution3dInferOutputShapeTest()
532 CHECK(expectedOutputShape == transposeConvolution2dLayer->InferOutputShapes(shapes).at(0)); in TransposeConvolution2dInferOutputShapeTest()
565 CHECK(expectedOutputShape == depthwiseConvolution2dLayer->InferOutputShapes(shapes).at(0)); in DepthwiseConvolution2dInferOutputShapeTest()
599 CHECK(expectedOutputShape == pooling3dLayer->InferOutputShapes(shapes).at(0)); in Pooling3dInferOutputShapeTest()
[all …]
|
| /external/armnn/src/armnn/layers/ |
| D | MergeLayer.cpp | 38 std::vector<TensorShape> inferredShapes = InferOutputShapes({ in ValidateTensorShapesFromInputs()
48 std::vector<TensorShape> MergeLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes)… in InferOutputShapes() function in armnn::MergeLayer
|
| D | ElementwiseUnaryLayer.cpp | 34 std::vector<TensorShape> ElementwiseUnaryLayer::InferOutputShapes(const std::vector<TensorShape>& i… in InferOutputShapes() function in armnn::ElementwiseUnaryLayer
51 std::vector<TensorShape> inferredShapes = InferOutputShapes({ in ValidateTensorShapesFromInputs()
|
| D | ShapeLayer.cpp | 45 …auto inferredShape = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape… in ValidateTensorShapesFromInputs()
52 std::vector<TensorShape> ShapeLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes)… in InferOutputShapes() function in armnn::ShapeLayer
|
| D | PermuteLayer.cpp | 38 std::vector<TensorShape> PermuteLayer::InferOutputShapes(const std::vector<TensorShape>& inputShape… in InferOutputShapes() function in armnn::PermuteLayer
53 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | ReshapeLayer.cpp | 35 std::vector<TensorShape> ReshapeLayer::InferOutputShapes(const std::vector<TensorShape>& inputShape… in InferOutputShapes() function in armnn::ReshapeLayer
49 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | TransposeLayer.cpp | 38 std::vector<TensorShape> TransposeLayer::InferOutputShapes(const std::vector<TensorShape>& inputSha… in InferOutputShapes() function in armnn::TransposeLayer
53 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | SliceLayer.cpp | 45 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
52 std::vector<TensorShape> SliceLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes)… in InferOutputShapes() function in armnn::SliceLayer
|
| D | FullyConnectedLayer.cpp | 35 std::vector<TensorShape> FullyConnectedLayer::InferOutputShapes(const std::vector<TensorShape>& inp… in InferOutputShapes() function in armnn::FullyConnectedLayer
54 std::vector<TensorShape> inferredShapes = InferOutputShapes( in ValidateTensorShapesFromInputs()
|
| D | LogicalBinaryLayer.cpp | 34 std::vector<TensorShape> LogicalBinaryLayer::InferOutputShapes(const std::vector<TensorShape>& inpu… in InferOutputShapes() function in armnn::LogicalBinaryLayer
66 std::vector<TensorShape> inferredShapes = InferOutputShapes({ in ValidateTensorShapesFromInputs()
|
| D | ElementwiseBaseLayer.cpp | 23 std::vector<TensorShape> ElementwiseBaseLayer::InferOutputShapes(const std::vector<TensorShape>& in… in InferOutputShapes() function in armnn::ElementwiseBaseLayer
69 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | ResizeLayer.cpp | 39 std::vector<TensorShape> ResizeLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes… in InferOutputShapes() function in armnn::ResizeLayer
71 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | DepthToSpaceLayer.cpp | 41 std::vector<TensorShape> DepthToSpaceLayer::InferOutputShapes(const std::vector<TensorShape>& input… in InferOutputShapes() function in armnn::DepthToSpaceLayer
70 std::vector<TensorShape> inferredShapes = InferOutputShapes({ in ValidateTensorShapesFromInputs()
|
| D | SpaceToDepthLayer.cpp | 44 std::vector<TensorShape> SpaceToDepthLayer::InferOutputShapes(const std::vector<TensorShape>& input… in InferOutputShapes() function in armnn::SpaceToDepthLayer
72 std::vector<TensorShape> inferredShapes = InferOutputShapes({ in ValidateTensorShapesFromInputs()
|
| D | ArgMinMaxLayer.cpp | 37 std::vector<TensorShape> ArgMinMaxLayer::InferOutputShapes(const std::vector<TensorShape>& inputSha… in InferOutputShapes() function in armnn::ArgMinMaxLayer
82 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | SpaceToBatchNdLayer.cpp | 43 std::vector<TensorShape> SpaceToBatchNdLayer::InferOutputShapes(const std::vector<TensorShape>& inp… in InferOutputShapes() function in armnn::SpaceToBatchNdLayer
78 std::vector<TensorShape> inferredShapes = InferOutputShapes({ in ValidateTensorShapesFromInputs()
|
| D | BatchToSpaceNdLayer.cpp | 53 …auto inferredShapes = InferOutputShapes({GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape… in ValidateTensorShapesFromInputs()
60 std::vector<TensorShape> BatchToSpaceNdLayer::InferOutputShapes(const std::vector<TensorShape>& inp… in InferOutputShapes() function in armnn::BatchToSpaceNdLayer
|
| D | StackLayer.cpp | 35 std::vector<TensorShape> StackLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes)… in InferOutputShapes() function in armnn::StackLayer
91 auto inferredShapes = InferOutputShapes(inputShapes); in ValidateTensorShapesFromInputs()
|
| D | PadLayer.cpp | 42 std::vector<TensorShape> PadLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) c… in InferOutputShapes() function in armnn::PadLayer
69 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | ComparisonLayer.cpp | 36 std::vector<TensorShape> ComparisonLayer::InferOutputShapes(const std::vector<TensorShape>& inputSh… in InferOutputShapes() function in armnn::ComparisonLayer
81 std::vector<TensorShape> inferredShapes = InferOutputShapes({ in ValidateTensorShapesFromInputs()
|
| D | ElementwiseBinaryLayer.cpp | 31 std::vector<TensorShape> ElementwiseBinaryLayer::InferOutputShapes(const std::vector<TensorShape>& … in InferOutputShapes() function in armnn::ElementwiseBinaryLayer
77 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
|
| D | PreluLayer.cpp | 38 std::vector<TensorShape> PreluLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes)… in InferOutputShapes() function in armnn::PreluLayer
108 std::vector<TensorShape> inferredShapes = InferOutputShapes( in ValidateTensorShapesFromInputs()
|
| D | GatherLayer.cpp | 34 std::vector<TensorShape> GatherLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes… in InferOutputShapes() function in armnn::GatherLayer
82 std::vector<TensorShape> inferredShapes = InferOutputShapes( in ValidateTensorShapesFromInputs()
|
| D | DetectionPostProcessLayer.cpp | 52 std::vector<TensorShape> inferredShapes = InferOutputShapes( in ValidateTensorShapesFromInputs()
80 std::vector<TensorShape> DetectionPostProcessLayer::InferOutputShapes(const std::vector<TensorShape… in InferOutputShapes() function in armnn::DetectionPostProcessLayer
|
| /external/armnn/delegate/classic/src/ |
| D | MultiLayerFacade.hpp | 86 virtual std::vector<armnn::TensorShape> InferOutputShapes( in InferOutputShapes() function in armnnDelegate::MultiLayerFacade
91 return m_FirstLayer->InferOutputShapes(inputShapes); in InferOutputShapes()
|