19 #include "ngraph/coordinate_diff.hpp"
20 #include "ngraph/op/op.hpp"
21 #include "ngraph/op/util/attr_types.hpp"
33 static constexpr NodeTypeInfo type_info{
"DeformableConvolution", 1};
34 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
59 const Output<Node>& deformable_values,
60 const Output<Node>& filters,
61 const Strides& strides,
62 const CoordinateDiff& pads_begin,
63 const CoordinateDiff& pads_end,
64 const Strides& dilations,
65 const PadType& auto_pad = PadType::EXPLICIT,
66 const int64_t group = 1,
67 const int64_t deformable_group = 1);
72 const Strides& get_strides()
const {
return m_strides; }
73 void set_strides(
const Strides& strides) { m_strides = strides; }
74 const Strides& get_dilations()
const {
return m_dilations; }
75 void set_dilations(
const Strides& dilations) { m_dilations = dilations; }
76 const CoordinateDiff& get_pads_begin()
const {
return m_pads_begin; }
77 void set_pads_begin(
const CoordinateDiff& pads_begin) { m_pads_begin = pads_begin; }
78 const CoordinateDiff& get_pads_end()
const {
return m_pads_end; }
79 void set_pads_end(
const CoordinateDiff& pads_end) { m_pads_end = pads_end; }
80 const PadType& get_auto_pad()
const {
return m_auto_pad; }
81 void set_auto_pad(
const PadType& auto_pad) { m_auto_pad = auto_pad; }
82 int64_t get_group()
const {
return m_group; }
83 void set_group(
const int64_t group) { m_group = group; }
84 int64_t get_deformable_group()
const {
return m_deformable_group; }
85 void set_deformable_group(
const int64_t deformable_group)
87 m_deformable_group = deformable_group;
89 virtual std::shared_ptr<Node>
90 clone_with_new_inputs(
const OutputVector& new_args)
const override;
95 CoordinateDiff m_pads_begin;
96 CoordinateDiff m_pads_end;
99 int64_t m_deformable_group;