21 #include "ngraph/node.hpp"
22 #include "ngraph/op/util/fused_op.hpp"
24 NGRAPH_SUPPRESS_DEPRECATED_START
36 static constexpr
NodeTypeInfo type_info{
"ShuffleChannels", 0};
50 const int64_t axis = 1,
51 const int64_t group = 1);
54 size_t get_zero_based_axis()
const;
58 virtual std::shared_ptr<Node>
59 clone_with_new_inputs(
const OutputVector& new_args)
const override;
61 int64_t get_axis()
const {
return m_axis; }
62 int64_t get_group()
const {
return m_group; }
64 const HostTensorVector& inputs)
const override;
71 Shape get_pre_shuffle_shape(
const Shape& data_shape)
const;
72 bool evaluate_shuffle_channels(
const HostTensorVector& outputs,
73 const HostTensorVector& inputs)
const;
83 NGRAPH_SUPPRESS_DEPRECATED_END
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Shape for a tensor.
Definition: shape.hpp:31
Root of all actual ops.
Definition: op.hpp:29
Permutes data in the channel dimension of the input.
Definition: shuffle_channels.hpp:34
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
const NodeTypeInfo & get_type_info() const override
Definition: shuffle_channels.hpp:37
virtual void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
ShuffleChannels(const Output< Node > &data, const int64_t axis=1, const int64_t group=1)
Constructs a ShuffleChannels node.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28