Commit 1c2e5b7a authored by Amy Zhuang's avatar Amy Zhuang Committed by Scott Cyphers

Add TBB flow graphs to DEX. (#1247)

* Add TBB flow graphs to DEX.

* Make edges from dummy start node to head nodes when traversing nodes.

* Use static_cast to cast TBB graph node.
Undefine __TBB_PREVIEW_LIGHTWEIGHT_POLICY.

* Code formatting.

* Remove clang wreserved-id-macro warning.
parent 8ad38f2e
...@@ -166,6 +166,8 @@ namespace ngraph ...@@ -166,6 +166,8 @@ namespace ngraph
std::list<std::function<void(CPURuntimeContext*)>> functors; std::list<std::function<void(CPURuntimeContext*)>> functors;
std::list<std::pair<std::function<bool(CPURuntimeContext*)>, size_t>> enables; std::list<std::pair<std::function<bool(CPURuntimeContext*)>, size_t>> enables;
std::list<std::pair<std::function<bool(CPURuntimeContext*)>, std::string>>
enable_nodename_list;
std::function<void(CPURuntimeContext*, std::vector<void*>&, std::vector<void*>&)> std::function<void(CPURuntimeContext*, std::vector<void*>&, std::vector<void*>&)>
executor; executor;
std::unordered_map<std::string, void*> tensor_data; std::unordered_map<std::string, void*> tensor_data;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment