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
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; }
88 const HostTensorVector& inputs)
const override;
92 bool evaluate_pad(
const HostTensorVector& outputs,
93 const HostTensorVector& inputs)
const;
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
Root of all actual ops.
Definition: op.hpp:29
Generic padding operation.
Definition: pad.hpp:31
PadMode get_pad_mode() const
Definition: pad.hpp:85
size_t get_version() const override
Definition: pad.hpp:72
Pad()=default
Constructs a generic padding operation.
CoordinateDiff get_pads_end() const
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.
CoordinateDiff get_pads_begin() const
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
const NodeTypeInfo & get_type_info() const override
Definition: pad.hpp:34
Pad(const Output< Node > &arg, const Output< Node > &pads_begin, const Output< Node > &pads_end, PadMode pad_mode)
Constructs a generic padding operation.
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
PadMode
Modes for the Pad operator.
Definition: attr_types.hpp:32
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28