Commit 04be484a authored by Robert Kimball's avatar Robert Kimball Committed by Michał Karzyński

Create tensor for the primary backend (#2970)

* create tensor for the primary backend

* move private objects to protected
parent fde204b0
...@@ -31,13 +31,13 @@ shared_ptr<runtime::Tensor> ...@@ -31,13 +31,13 @@ shared_ptr<runtime::Tensor>
runtime::hybrid::HybridBackend::create_tensor(const element::Type& element_type, runtime::hybrid::HybridBackend::create_tensor(const element::Type& element_type,
const Shape& shape) const Shape& shape)
{ {
return make_shared<HybridTensor>(element_type, shape); return m_backend_list[0]->create_tensor(element_type, shape);
} }
shared_ptr<runtime::Tensor> runtime::hybrid::HybridBackend::create_tensor( shared_ptr<runtime::Tensor> runtime::hybrid::HybridBackend::create_tensor(
const element::Type& element_type, const Shape& shape, void* memory_pointer) const element::Type& element_type, const Shape& shape, void* memory_pointer)
{ {
return make_shared<HybridTensor>(element_type, shape, memory_pointer); return m_backend_list[0]->create_tensor(element_type, shape, memory_pointer);
} }
shared_ptr<runtime::Executable> shared_ptr<runtime::Executable>
......
...@@ -54,7 +54,7 @@ public: ...@@ -54,7 +54,7 @@ public:
bool is_supported(const ngraph::Node& node) const override; bool is_supported(const ngraph::Node& node) const override;
void set_debug_enabled(bool flag) { m_debug_enabled = flag; } void set_debug_enabled(bool flag) { m_debug_enabled = flag; }
private: protected:
std::vector<std::shared_ptr<runtime::Backend>> m_backend_list; std::vector<std::shared_ptr<runtime::Backend>> m_backend_list;
bool m_debug_enabled = false; bool m_debug_enabled = false;
}; };
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