diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/nnapi')
| -rw-r--r-- | runtimes/pure_arm_compute/src/internal/nnapi/tensor/Reader.h | 2 | ||||
| -rw-r--r-- | runtimes/pure_arm_compute/src/internal/nnapi/tensor/View.h | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/nnapi/tensor/Reader.h b/runtimes/pure_arm_compute/src/internal/nnapi/tensor/Reader.h index cc51db594..fc6d490da 100644 --- a/runtimes/pure_arm_compute/src/internal/nnapi/tensor/Reader.h +++ b/runtimes/pure_arm_compute/src/internal/nnapi/tensor/Reader.h @@ -49,7 +49,7 @@ public: Reader(const ::nnfw::misc::tensor::Shape &shape, const T *ptr, size_t len) : _shape{shape}, _ptr{ptr} { - assert(shape.element_nums() * sizeof(T) == len); + assert(shape.num_elements() * sizeof(T) == len); initialize(); } diff --git a/runtimes/pure_arm_compute/src/internal/nnapi/tensor/View.h b/runtimes/pure_arm_compute/src/internal/nnapi/tensor/View.h index f8f297f97..4766851b9 100644 --- a/runtimes/pure_arm_compute/src/internal/nnapi/tensor/View.h +++ b/runtimes/pure_arm_compute/src/internal/nnapi/tensor/View.h @@ -47,7 +47,7 @@ public: // NOTE The parameter len denotes the number of bytes. View(const ::nnfw::misc::tensor::Shape &shape, T *ptr, size_t len) : _shape{shape}, _ptr{ptr} { - assert(shape.element_nums() * sizeof(T) == len); + assert(shape.num_elements() * sizeof(T) == len); } public: |