[RUNTIME] Add clear() function in tvm::Map class

apache · Apr 12, 2021 · 3ca26b6 · 3ca26b6
1 parent f38ae65
commit 3ca26b6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/include/tvm/runtime/container.h b/include/tvm/runtime/container.h
@@ -2980,6 +2980,15 @@ class Map : public ObjectRef {
   }
   /*! \return whether array is empty */
   bool empty() const { return size() == 0; }
+  /*! \brief Release reference to all the elements */
+  void clear() {
+    MapNode* n = GetMapNode();
+    if (n != nullptr) {
+      for (auto it = n->begin(); it != n->end(); ++it) {
+        CopyOnWrite()->erase(it);
+      }
+    }
+  }
   /*!
    * \brief set the Map.
    * \param key The index key.