19 #include "ngraph/coordinate_diff.hpp"
20 #include "ngraph/op/op.hpp"
21 #include "ngraph/op/util/attr_types.hpp"
30 class NGRAPH_API
Pad :
public Op
33 static constexpr NodeTypeInfo type_info{
"Pad", 1};
34 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
49 Pad(
const Output<Node>& arg,
50 const Output<Node>& pads_begin,
51 const Output<Node>& pads_end,
52 const Output<Node>& arg_pad_value,
63 Pad(
const Output<Node>& arg,
64 const Output<Node>& pads_begin,
65 const Output<Node>& pads_end,
74 virtual std::shared_ptr<Node>
75 clone_with_new_inputs(
const OutputVector& new_args)
const override;
86 void set_pad_mode(
PadMode pad_mode) { m_pad_mode = pad_mode; }
87 bool evaluate(
const HostTensorVector& outputs,
88 const HostTensorVector& inputs)
const override;
Generic padding operation.
Definition: pad.hpp:31
CoordinateDiff get_pads_begin() const
size_t get_version() const override
Definition: pad.hpp:72
PadMode get_pad_mode() const
Definition: pad.hpp:85
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Pad(const Output< Node > &arg, const Output< Node > &pads_begin, const Output< Node > &pads_end, PadMode pad_mode)
Constructs a generic padding operation.
const NodeTypeInfo & get_type_info() const override
Definition: pad.hpp:34
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Pad(const Output< Node > &arg, const Output< Node > &pads_begin, const Output< Node > &pads_end, const Output< Node > &arg_pad_value, PadMode pad_mode)
Constructs a generic padding operation.
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
CoordinateDiff get_pads_end() const
PadMode
Modes for the Pad operator.
Definition: attr_types.hpp:32
Pad()=default
Constructs a generic padding operation.
Root of all actual ops.
Definition: op.hpp:29