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