Commit d52d676b authored by Sang Ik Lee's avatar Sang Ik Lee Committed by Scott Cyphers

Fix bug introduced by #2238 on master (#2514)

* Fix bug introduced by #2238

* Apply style.
parent 6123134a
......@@ -1918,6 +1918,8 @@ string runtime::cpu::CPU_ExternalFunction::strip_comments(const string& s)
return out.str();
}
#endif
std::unordered_set<descriptor::Tensor*>&
runtime::cpu::CPU_ExternalFunction::get_tensor_set(descriptor::Tensor* output_tensor)
{
......@@ -1928,5 +1930,3 @@ std::unordered_set<descriptor::Tensor*>&
NGRAPH_ASSERT(output_buffer_it != bufferID_to_tensorSets.end());
return output_buffer_it->second.second;
}
#endif
......@@ -203,9 +203,6 @@ namespace ngraph
std::string emit_op_as_function(const Node&, const std::string& function_name);
std::string strip_comments(const std::string&);
std::unordered_set<descriptor::Tensor*>&
get_tensor_set(descriptor::Tensor* output_tensor);
std::unique_ptr<codegen::Compiler> m_compiler;
std::unique_ptr<codegen::ExecutionEngine> m_execution_engine;
......@@ -216,6 +213,8 @@ namespace ngraph
// so they don't get freed before we are done with them
std::vector<std::shared_ptr<Node>> m_active_constants;
#endif
std::unordered_set<descriptor::Tensor*>&
get_tensor_set(descriptor::Tensor* output_tensor);
std::shared_ptr<ngraph::Function> m_function;
bool m_release_function;
......
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