19 #include "ngraph/node.hpp"
20 #include "ngraph/op/util/fused_op.hpp"
42 static constexpr NodeTypeInfo type_info{
"SpaceToBatch", 1};
43 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
55 const Output<Node>& block_shape,
56 const Output<ngraph::Node>& pads_begin,
57 const Output<ngraph::Node>& pads_end);
61 clone_with_new_inputs(
const OutputVector& new_args)
const override;
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
SpaceToBatch(const Output< Node > &data, const Output< Node > &block_shape, const Output< ngraph::Node > &pads_begin, const Output< ngraph::Node > &pads_end)
Constructs a SpaceToBatch operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
SpaceToBatch permutes data tensor blocks of spatial data into batch dimension.
Definition: space_to_batch.hpp:40
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
const NodeTypeInfo & get_type_info() const override
Definition: space_to_batch.hpp:43
Root of all actual ops.
Definition: op.hpp:29