Unverified Commit ab3e3965 authored by Nick Korovaiko's avatar Nick Korovaiko Committed by GitHub

Remove references on element::Type (m_element_type) in tensor.hpp,types.hpp,convert.hpp (#389)

* remove refs on types in tensor,tensor_view_type,convert

* fix build breaks
parent 46199d5f
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
#include <iostream> #include <iostream>
#include <memory> #include <memory>
#include <vector> #include <vector>
#include "ngraph/types/type.hpp"
namespace ngraph namespace ngraph
{ {
...@@ -65,7 +66,7 @@ public: ...@@ -65,7 +66,7 @@ public:
static std::string make_tensor_name(const Node* node, size_t value_index); static std::string make_tensor_name(const Node* node, size_t value_index);
void set_is_output() { m_is_output = true; } void set_is_output() { m_is_output = true; }
protected: protected:
const element::Type& m_element_type; const element::Type m_element_type;
PrimaryTensorView* m_primary_tensor_view; PrimaryTensorView* m_primary_tensor_view;
bool m_is_output; bool m_is_output;
bool m_is_input; bool m_is_input;
......
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
#pragma once #pragma once
#include "ngraph/ops/op.hpp" #include "ngraph/ops/op.hpp"
#include "ngraph/types/type.hpp"
namespace ngraph namespace ngraph
{ {
...@@ -63,7 +64,7 @@ namespace ngraph ...@@ -63,7 +64,7 @@ namespace ngraph
bool is_functionally_identical(const Node&) const override; bool is_functionally_identical(const Node&) const override;
protected: protected:
const ngraph::element::Type& m_element_type; const ngraph::element::Type m_element_type;
virtual void generate_adjoints(autodiff::Adjoints& adjoints, virtual void generate_adjoints(autodiff::Adjoints& adjoints,
const std::shared_ptr<Node>& delta) override; const std::shared_ptr<Node>& delta) override;
}; };
......
...@@ -69,7 +69,7 @@ namespace ngraph ...@@ -69,7 +69,7 @@ namespace ngraph
friend std::ostream& operator<<(std::ostream&, const TensorViewType&); friend std::ostream& operator<<(std::ostream&, const TensorViewType&);
protected: protected:
const element::Type& m_element_type; const element::Type m_element_type;
Shape m_shape; Shape m_shape;
}; };
......
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