19 #include "ngraph/op/op.hpp"
30 static constexpr
NodeTypeInfo type_info{
"DeformablePSROIPooling", 1};
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);
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;
97 int64_t m_group_size = 1;
98 std::string m_mode =
"bilinear";
99 int64_t m_spatial_bins_x = 1;
100 int64_t m_spatial_bins_y = 1;
101 float m_trans_std = 1.f;
102 int64_t m_part_size = 1;
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28