19 #include "ngraph/op/op.hpp"
30 static constexpr
NodeTypeInfo type_info{
"ROIPooling", 0};
42 const Shape& output_size,
43 const float spatial_scale,
44 const std::string& method =
"max");
48 virtual std::shared_ptr<Node>
49 clone_with_new_inputs(
const OutputVector& new_args)
const override;
51 const Shape& get_output_size()
const {
return m_output_size; }
52 float get_spatial_scale()
const {
return m_spatial_scale; }
53 const std::string& get_method()
const {
return m_method; }
57 Shape m_output_size{0, 0};
58 float m_spatial_scale;
59 std::string m_method =
"max";
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
Shape for a tensor.
Definition: shape.hpp:31
Root of all actual ops.
Definition: op.hpp:29
Definition: roi_pooling.hpp:28
const NodeTypeInfo & get_type_info() const override
Definition: roi_pooling.hpp:31
ROIPooling(const Output< Node > &input, const Output< Node > &coords, const Shape &output_size, const float spatial_scale, const std::string &method="max")
Constructs a ROIPooling 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