Unverified Commit f62f31e6 authored by Scott Cyphers's avatar Scott Cyphers Committed by GitHub

Parameter attribute setters (#3421)

parent 14624c03
...@@ -26,8 +26,7 @@ const string op::Parameter::type_name{"Parameter"}; ...@@ -26,8 +26,7 @@ const string op::Parameter::type_name{"Parameter"};
op::Parameter::Parameter(const element::Type& element_type, op::Parameter::Parameter(const element::Type& element_type,
const PartialShape& pshape, const PartialShape& pshape,
const bool cacheable) const bool cacheable)
: Op(NodeVector{}) : m_cacheable(cacheable)
, m_cacheable(cacheable)
, m_partial_shape(pshape) , m_partial_shape(pshape)
, m_element_type(element_type) , m_element_type(element_type)
, m_is_relevant_to_shapes(false) , m_is_relevant_to_shapes(false)
......
...@@ -38,6 +38,7 @@ namespace ngraph ...@@ -38,6 +38,7 @@ namespace ngraph
NGRAPH_API NGRAPH_API
static const std::string type_name; static const std::string type_name;
const std::string& description() const override { return type_name; } const std::string& description() const override { return type_name; }
/// \brief Constructions a tensor-typed parameter node.
Parameter() = default; Parameter() = default;
/// \brief Constructions a tensor-typed parameter node. /// \brief Constructions a tensor-typed parameter node.
/// ///
...@@ -57,6 +58,19 @@ namespace ngraph ...@@ -57,6 +58,19 @@ namespace ngraph
bool is_relevant_to_shapes() const; bool is_relevant_to_shapes() const;
void set_is_relevant_to_shapes(bool is_relevant); void set_is_relevant_to_shapes(bool is_relevant);
const PartialShape& get_partial_shape() const { return m_partial_shape; }
PartialShape& get_partial_shape() { return m_partial_shape; }
void set_partial_shape(const PartialShape& partial_shape)
{
m_partial_shape = partial_shape;
}
const element::Type& get_element_type() const { return m_element_type; }
void set_element_type(const element::Type& element_type)
{
m_element_type = element_type;
}
protected: protected:
bool m_cacheable; bool m_cacheable;
PartialShape m_partial_shape; PartialShape m_partial_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