fix example and impl

apache · Nov 28, 2019 · 29e7c6a · 29e7c6a
1 parent 0ba0aa8
commit 29e7c6a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/tvm/runtime/container.h b/include/tvm/runtime/container.h
@@ -55,8 +55,10 @@ namespace runtime {
  *  void Init(Iterator begin, Iterator end) {
  *   size_t num_elems = std::distance(begin, end);
  *   auto it = begin;
+ *   this->size = 0;
  *   for (size_t i = 0; i < num_elems; ++i) {
  *     InplaceArrayBase::EmplaceInit(i, *it++);
+ *     this->size++;
  *   }
  *  }
  * }
@@ -223,7 +225,6 @@ class ADT : public ObjectRef {
     size_t num_elems = std::distance(begin, end);
     auto ptr = make_inplace_array_object<ADTObj, ObjectRef>(num_elems);
     ptr->tag = tag;
-    ptr->size = num_elems;
     ptr->Init(begin, end);
     data_ = std::move(ptr);
   }