Commit c9c01be1 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

CPU: Add primitive vector getter

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