19 #include "ngraph/op/op.hpp"
32 enum class PoolingMode
56 const int sampling_ratio,
57 const float spatial_scale,
58 const std::string& mode);
65 const int sampling_ratio,
66 const float spatial_scale,
67 const PoolingMode mode);
71 virtual std::shared_ptr<Node>
72 clone_with_new_inputs(
const OutputVector& new_args)
const override;
74 int get_pooled_h()
const {
return m_pooled_h; }
75 int get_pooled_w()
const {
return m_pooled_w; }
76 int get_sampling_ratio()
const {
return m_sampling_ratio; }
77 float get_spatial_scale()
const {
return m_spatial_scale; }
78 PoolingMode get_mode()
const {
return m_mode; }
80 const HostTensorVector& inputs)
const override;
83 PoolingMode mode_from_string(
const std::string& mode)
const;
89 float m_spatial_scale;
96 std::ostream& operator<<(std::ostream& s,
const op::v3::ROIAlign::PoolingMode& mode);
109 "AttributeAdapter<op::v3::ROIAlign::PoolingMode>", 3};
110 const DiscreteTypeInfo& get_type_info()
const override {
return type_info; }
An AttributeAdapter "captures" an attribute as an AT& and makes it available as a ValueAccessor<VAT>.
Definition: attribute_adapter.hpp:171
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
Access an enum via a string.
Definition: attribute_adapter.hpp:178
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
Definition: roi_align.hpp:28
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
virtual void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
ROIAlign(const Output< Node > &input, const Output< Node > &rois, const Output< Node > &batch_indices, const int pooled_h, const int pooled_w, const int sampling_ratio, const float spatial_scale, const std::string &mode)
Constructs a ROIAlign node matching the ONNX ROIAlign specification.
const NodeTypeInfo & get_type_info() const override
Definition: roi_align.hpp:31
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28