diff options
Diffstat (limited to 'src/armnn/layers/PermuteLayer.hpp')
| -rw-r--r-- | src/armnn/layers/PermuteLayer.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/layers/PermuteLayer.hpp b/src/armnn/layers/PermuteLayer.hpp index c060a16390..2700dd2c7b 100644 --- a/src/armnn/layers/PermuteLayer.hpp +++ b/src/armnn/layers/PermuteLayer.hpp @@ -18,6 +18,7 @@ public: PermuteLayer* Clone(Graph& graph) const override; void ValidateTensorShapesFromInputs() override; + std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override; const PermutationVector& GetPermutation() const { |