[neurun] Add basic methods to LayoutSet (#2623)
author이한종/동작제어Lab(SR)/Engineer/삼성전자 <hanjoung.lee@samsung.com>
Thu, 6 Sep 2018 08:24:01 +0000 (17:24 +0900)
committer이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Thu, 6 Sep 2018 08:24:01 +0000 (17:24 +0900)
Introduce `contains` and `size` method for user queries.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>
runtimes/neurun/src/graph/operand/LayoutSet.h

index 27a5c7f..1468b28 100644 (file)
@@ -22,6 +22,8 @@ public:
 public:
   void add(const Layout &layout) { _set.insert(layout); }
   void remove(const Layout &layout) { _set.erase(layout); }
+  uint32_t size() const { return static_cast<uint32_t>(_set.size()); }
+  bool contains(const Layout &layout) const { return _set.find(layout) != _set.end(); }
 
 public:
   std::unordered_set<Layout>::const_iterator begin() const { return _set.begin(); }