Imported Upstream version 1.15.0upstream/1.15.0submit/tizen/20210427.093759submit/tizen/20210423.055448submit/tizen/20210422.015846submit/tizen/20210421.062230accepted/tizen/unified/20210428.040443

1 files changed, 10 insertions, 10 deletions

diff --git a/runtime/onert/backend/cpu/ops/TransposeLayer.cc b/runtime/onert/backend/cpu/ops/TransposeLayer.cc
index 3362c3396..850c07ab8 100644
--- a/runtime/onert/backend/cpu/ops/TransposeLayer.cc
+++ b/runtime/onert/backend/cpu/ops/TransposeLayer.cc
@@ -38,33 +38,33 @@ TransposeLayer::TransposeLayer() : _input(nullptr), _perm(nullptr), _output(null
 template <typename T> void TransposeLayer::transpose()
 {
   nnfw::cker::TransposeParams param;
-  assert(_perm->num_dimensions() == 1);
+  auto perm_shape = _perm->getShape();
+  assert(perm_shape.rank() == 1);
 
-  param.perm_count = _input->num_dimensions();
-  if (_perm->dimension(0) == 0) // This means _perm is (n-1...0)
+  param.perm_count = _input->getShape().rank();
+  if (perm_shape.dim(0) == 0) // This means _perm is (n-1...0)
   {
     const auto begin = param.perm;
-    const auto end = param.perm + _input->num_dimensions();
+    const auto end = param.perm + _input->getShape().rank();
     std::iota(begin, end, 0);
     std::reverse(begin, end);
   }
   else
   {
-    assert(param.perm_count == static_cast<int>(_perm->dimension(0)));
+    assert(param.perm_count == static_cast<int>(perm_shape.dim(0)));
     for (auto i = 0; i < param.perm_count; i++)
     {
-      param.perm[i] = *(reinterpret_cast<const int32_t *>(_perm->buffer()) + i);
+      param.perm[i] = *(getBuffer<int32_t>(_perm) + i);
     }
   }
 
-  nnfw::cker::Transpose(param, getTensorShape(_input),
-                        reinterpret_cast<const T *>(_input->buffer()), getTensorShape(_output),
-                        reinterpret_cast<T *>(_output->buffer()));
+  nnfw::cker::Transpose(param, getShape(_input), getBuffer<T>(_input), getShape(_output),
+                        getBuffer<T>(_output));
 }
 
 void TransposeLayer::transposeQuant8()
 {
-  if (_input->data_offset() != _output->data_offset())
+  if (_input->data_zero_point() != _output->data_zero_point())
   {
     throw std::runtime_error("TransposeLayer : qassym8 input and output offsets unmatched");
   }