19 #include "ngraph/node.hpp"
20 #include "ngraph/op/op.hpp"
21 #include "ngraph/op/util/attr_types.hpp"
22 #include "ngraph/op/util/fused_op.hpp"
23 #include "ngraph/runtime/host_tensor.hpp"
25 NGRAPH_SUPPRESS_DEPRECATED_START
44 NGRAPH_RTTI_DECLARATION;
46 enum class DepthToSpaceMode
62 const DepthToSpaceMode& mode,
63 std::size_t block_size = 1);
66 const std::string& mode,
67 std::size_t block_size = 1);
70 std::size_t get_block_size()
const {
return m_blocksize; }
71 DepthToSpaceMode get_mode()
const {
return m_mode; }
72 virtual std::shared_ptr<Node>
73 clone_with_new_inputs(
const OutputVector& new_args)
const override;
76 const HostTensorVector& inputs)
const override;
79 std::size_t m_blocksize;
80 DepthToSpaceMode m_mode;
81 DepthToSpaceMode mode_from_string(
const std::string& mode)
const;
84 bool evaluate_depth_to_space(
const HostTensorVector& outputs,
85 const HostTensorVector& inputs)
const;
92 std::ostream& operator<<(std::ostream& s,
const op::v0::DepthToSpace::DepthToSpaceMode& type);
105 "AttributeAdapter<op::v0::DepthToSpace::DepthToSpaceMode>", 0};
106 const DiscreteTypeInfo& get_type_info()
const override {
return type_info; }
110 NGRAPH_SUPPRESS_DEPRECATED_END
An AttributeAdapter "captures" an attribute as an AT& and makes it available as a ValueAccessor<VAT>.
Definition: attribute_adapter.hpp:171
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
Access an enum via a string.
Definition: attribute_adapter.hpp:178
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
DepthToSpace permutes data from the depth dimension of the input blob into spatial dimensions.
Definition: depth_to_space.hpp:42
DepthToSpace(const Output< Node > &data, const DepthToSpaceMode &mode, std::size_t block_size=1)
Constructs a DepthToSpace operation.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28