Commit c9c01be1 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

CPU: Add primitive vector getter

parent 154daf06
...@@ -22,6 +22,11 @@ ...@@ -22,6 +22,11 @@
using namespace ngraph::runtime::cpu; using namespace ngraph::runtime::cpu;
const std::vector<mkldnn::primitive*>& MKLDNNEmitter::get_mkldnn_primitives() const
{
return mkldnn_primitives;
}
size_t MKLDNNEmitter::insert_primitive(mkldnn::primitive* primitive) size_t MKLDNNEmitter::insert_primitive(mkldnn::primitive* primitive)
{ {
mkldnn_primitives.emplace_back(primitive); mkldnn_primitives.emplace_back(primitive);
......
...@@ -39,6 +39,8 @@ namespace ngraph ...@@ -39,6 +39,8 @@ namespace ngraph
{ {
} }
const std::vector<mkldnn::primitive*>& get_mkldnn_primitives() const;
size_t insert_primitive(mkldnn::primitive* primitive); size_t insert_primitive(mkldnn::primitive* primitive);
const std::vector<size_t>& get_primitive_deps(size_t index) const; const std::vector<size_t>& get_primitive_deps(size_t index) const;
......
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