diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-21 16:48:17 +0000 |
---|---|---|
committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-21 17:13:30 +0000 |
commit | a0d2844d05dd9ae15733d426b04ab651457604ce (patch) | |
tree | 7ae0787d7be49ec2b1bb754ff35e13e8b71c73d2 /src/armnn/layers/StridedSliceLayer.hpp | |
parent | 3d93bc47f42b339d82cfcf56a90c9264dd46d70a (diff) | |
download | armnn-a0d2844d05dd9ae15733d426b04ab651457604ce.tar.gz |
IVGCVSW-2086 Update StridedSliceLayer and StridedSliceDescriptor
Change-Id: Ifa88a879dd239f60ab27330d6b73859393828ef0
Diffstat (limited to 'src/armnn/layers/StridedSliceLayer.hpp')
-rw-r--r-- | src/armnn/layers/StridedSliceLayer.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/layers/StridedSliceLayer.hpp b/src/armnn/layers/StridedSliceLayer.hpp index 33a44243a5..c3aad53e19 100644 --- a/src/armnn/layers/StridedSliceLayer.hpp +++ b/src/armnn/layers/StridedSliceLayer.hpp @@ -17,6 +17,8 @@ public: StridedSliceLayer* Clone(Graph& graph) const override; + std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override; + void ValidateTensorShapesFromInputs() override; protected: @@ -24,4 +26,4 @@ protected: ~StridedSliceLayer() = default; }; -} // namespace
\ No newline at end of file +} // namespace |