22 #include "ngraph/partial_shape.hpp"
23 #include "ngraph/shape.hpp"
24 #include "ngraph/type/element_type.hpp"
40 const PartialShape& pshape,
41 const std::string& name);
43 const PartialShape& pshape,
45 size_t node_output_number);
47 const std::string& get_name()
const;
48 void set_name(
const std::string& name);
49 void set_tensor_type(
const element::Type& element_type,
const PartialShape& pshape);
51 void set_partial_shape(
const PartialShape& partial_shape);
53 const element::Type& get_element_type()
const {
return m_element_type; }
54 const Shape& get_shape()
const;
55 const PartialShape& get_partial_shape()
const {
return m_partial_shape; }
56 void set_pool_offset(
size_t);
57 size_t get_pool_offset()
const;
69 PartialShape m_partial_shape;
70 Node* m_node{
nullptr};
71 size_t m_node_output_number{0};
74 size_t m_pool_offset{0};