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};
37 const NodeTypeInfo& get_type_info()
const override {
return type_info; }
50 const int64_t axis = 1,
51 const int64_t group = 1);
54 size_t get_zero_based_axis()
const;
56 virtual void pre_validate_and_infer_types()
override;
58 virtual OutputVector decompose_op()
const override;
60 virtual std::shared_ptr<Node>
61 clone_with_new_inputs(
const OutputVector& new_args)
const override;
63 int64_t get_axis()
const {
return m_axis; }
64 int64_t get_group()
const {
return m_group; }
70 Shape get_pre_shuffle_shape(
const Shape& data_shape)
const;
76 using v0::ShuffleChannels;
80 NGRAPH_SUPPRESS_DEPRECATED_END