Update tensor reshape check when buffer type is image.

7866f724 · liuqi · 9e52599e · 7866f724
隐藏空白更改
内联并排

Showing with 5 addition and 1 deletion

mace/core/tensor.h mace/core/tensor.h +5 -1

未找到文件。
--- a/mace/core/tensor.h
+++ b/mace/core/tensor.h
@@ -217,7 +217,11 @@ class Tensor {
  inline void Reshape(const std::vector<index_t> &shape) {
    shape_ = shape;
-    MACE_CHECK(raw_size() <= buffer_->size());
+    if (has_opencl_image()) {
+      MACE_CHECK(raw_size() <= 4 * buffer_->size());
+    } else {
+      MACE_CHECK(raw_size() <= buffer_->size());
+    }
  }
  inline MaceStatus Resize(const std::vector<index_t> &shape) {