diff options
author | Hyun Sik Yoon <63768207+hyunsik-yoon@users.noreply.github.com> | 2020-06-25 15:42:51 +0900 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-06-25 15:42:51 +0900 |
commit | 3e705f8d597122a6c02d95311e6888e6da71b22b (patch) | |
tree | 92736e1a13ea1d6894d7f6bed8a87c83e1efbfab | |
parent | 1e03fa96c6a7173d4f60b7c4db16b9d7b44c6e65 (diff) | |
download | nnfw-3e705f8d597122a6c02d95311e6888e6da71b22b.tar.gz nnfw-3e705f8d597122a6c02d95311e6888e6da71b22b.tar.bz2 nnfw-3e705f8d597122a6c02d95311e6888e6da71b22b.zip |
[onert] int64 support for StridedSliceLayer.cc (#2615)
This adds int64 support for StridedSliceLayer.cc.
Signed-off-by: Hyun Sik Yoon <hyunsik.yoon.1024@gmail.com>
-rw-r--r-- | runtime/onert/backend/cpu/ops/StridedSliceLayer.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/runtime/onert/backend/cpu/ops/StridedSliceLayer.cc b/runtime/onert/backend/cpu/ops/StridedSliceLayer.cc index 07d1eaf2b..dcbb87734 100644 --- a/runtime/onert/backend/cpu/ops/StridedSliceLayer.cc +++ b/runtime/onert/backend/cpu/ops/StridedSliceLayer.cc @@ -78,6 +78,10 @@ void StridedSliceLayer::run() { stridedSliceImpl<int32_t>(); } + else if (_input->data_type() == OperandType::INT64) + { + stridedSliceImpl<int64_t>(); + } else { throw std::runtime_error{"StridedSlice: unsupported data type"}; |