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};
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;
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A difference (signed) of tensor element coordinates.
Definition: coordinate_diff.hpp:30
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Strides for a tensor.
Definition: strides.hpp:30
Root of all actual ops.
Definition: op.hpp:29
PadType
Padding Type used for Convolution and Pooling
Definition: attr_types.hpp:73
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28