diff options
Diffstat (limited to 'OutputShapeUtils.hpp')
-rw-r--r-- | OutputShapeUtils.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/OutputShapeUtils.hpp b/OutputShapeUtils.hpp index f314252f..6e2a76db 100644 --- a/OutputShapeUtils.hpp +++ b/OutputShapeUtils.hpp @@ -10,6 +10,8 @@ namespace armnn_driver { +bool IsDynamicOutput(const armnn::TensorInfo& outputInfo); + armnn::TensorShape InferPreluOutputShape(const armnn::TensorShape& inputShape, const armnn::TensorShape& alphaShape); } // namespace armnn_driver |