19 #include "ngraph/node.hpp"
20 #include "ngraph/op/util/fused_op.hpp"
21 #include "ngraph/runtime/host_tensor.hpp"
23 NGRAPH_SUPPRESS_DEPRECATED_START
41 static constexpr
NodeTypeInfo type_info{
"SpaceToDepth", 0};
43 enum class SpaceToDepthMode
59 const SpaceToDepthMode& mode,
60 std::size_t block_size = 1);
63 const std::string& mode,
64 std::size_t block_size = 1);
67 std::size_t get_block_size()
const {
return m_blocksize; }
68 SpaceToDepthMode get_mode()
const {
return m_mode; }
70 virtual std::shared_ptr<Node>
71 clone_with_new_inputs(
const OutputVector& new_args)
const override;
74 const HostTensorVector& inputs)
const override;
77 std::size_t m_blocksize;
78 SpaceToDepthMode m_mode;
81 bool evaluate_space_to_depth(
const HostTensorVector& outputs,
82 const HostTensorVector& inputs)
const;
89 std::ostream& operator<<(std::ostream& s,
const op::v0::SpaceToDepth::SpaceToDepthMode& type);
102 "AttributeAdapter<op::v0::SpaceToDepth::SpaceToDepthMode>", 0};
103 const DiscreteTypeInfo& get_type_info()
const override {
return type_info; }
107 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
SpaceToDepth permutes input tensor blocks of spatial data into depth dimension.
Definition: space_to_depth.hpp:39
SpaceToDepth(const Output< Node > &data, const SpaceToDepthMode &mode, std::size_t block_size=1)
Constructs a SpaceToDepth operation.
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
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: space_to_depth.hpp:42
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28