19 #include "ngraph/op/op.hpp"
30 static constexpr
NodeTypeInfo type_info{
"PSROIPooling", 0};
47 const size_t output_dim,
48 const size_t group_size,
49 const float spatial_scale,
52 const std::string& mode);
57 virtual std::shared_ptr<Node>
58 clone_with_new_inputs(
const OutputVector& new_args)
const override;
60 size_t get_output_dim()
const {
return m_output_dim; }
61 size_t get_group_size()
const {
return m_group_size; }
62 float get_spatial_scale()
const {
return m_spatial_scale; }
63 int get_spatial_bins_x()
const {
return m_spatial_bins_x; }
64 int get_spatial_bins_y()
const {
return m_spatial_bins_y; }
65 const std::string& get_mode()
const {
return m_mode; }
69 float m_spatial_scale;
75 using v0::PSROIPooling;
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
Definition: psroi_pooling.hpp:28
const NodeTypeInfo & get_type_info() const override
Definition: psroi_pooling.hpp:31
PSROIPooling(const Output< Node > &input, const Output< Node > &coords, const size_t output_dim, const size_t group_size, const float spatial_scale, int spatial_bins_x, int spatial_bins_y, const std::string &mode)
Constructs a PSROIPooling operation.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28