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,8 +217,12 @@ class Tensor {

  inline void Reshape(const std::vector<index_t> &shape) {
    shape_ = shape;
+    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) {
    shape_ = shape;