19 #include "ngraph/op/op.hpp"
30 static constexpr NodeTypeInfo type_info{
"DeformablePSROIPooling", 1};
31 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
57 const Output<Node>& coords,
58 const Output<Node>& offsets,
59 const int64_t output_dim,
60 const float spatial_scale,
61 const int64_t group_size = 1,
62 const std::string mode =
"bilinear_deformable",
63 int64_t spatial_bins_x = 1,
64 int64_t spatial_bins_y = 1,
66 int64_t part_size = 1);
69 const Output<Node>& coords,
70 const int64_t output_dim,
71 const float spatial_scale,
72 const int64_t group_size = 1,
73 const std::string mode =
"bilinear_deformable",
74 int64_t spatial_bins_x = 1,
75 int64_t spatial_bins_y = 1,
77 int64_t part_size = 1);
83 virtual std::shared_ptr<Node>
84 clone_with_new_inputs(
const OutputVector& new_args)
const override;
86 int64_t get_output_dim()
const {
return m_output_dim; }
87 int64_t get_group_size()
const {
return m_group_size; }
88 float get_spatial_scale()
const {
return m_spatial_scale; }
89 const std::string& get_mode()
const {
return m_mode; }
90 int64_t get_spatial_bins_x()
const {
return m_spatial_bins_x; }
91 int64_t get_spatial_bins_y()
const {
return m_spatial_bins_y; }
92 float get_trans_std()
const {
return m_trans_std; }
93 int64_t get_part_size()
const {
return m_part_size; }
96 float m_spatial_scale;
99 int64_t m_spatial_bins_x;
100 int64_t m_spatial_bins_y;
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
Root of all actual ops.
Definition: op.hpp:29