19 #include "ngraph/op/op.hpp"
30 static constexpr
NodeTypeInfo type_info{
"ExtractImagePatches", 3};
51 virtual std::shared_ptr<Node>
52 clone_with_new_inputs(
const OutputVector& new_args)
const override;
54 const Shape& get_sizes()
const {
return m_patch_sizes; }
55 void set_sizes(
const Shape& sizes) { m_patch_sizes = sizes; }
56 const Strides& get_strides()
const {
return m_patch_movement_strides; }
57 void set_strides(
const Strides& strides) { m_patch_movement_strides = strides; }
58 const Shape& get_rates()
const {
return m_patch_selection_rates; }
59 void set_rates(
const Shape& rates) { m_patch_selection_rates = rates; }
60 const PadType& get_auto_pad()
const {
return m_padding; }
61 void set_auto_pad(
PadType& padding) { m_padding = padding; }
64 Strides m_patch_movement_strides;
65 Shape m_patch_selection_rates;
69 using v3::ExtractImagePatches;
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
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