Public Member Functions | Data Fields
ngraph::pass::ShuffleChannelsFusion Class Reference

ShuffleChannelsFusion transformation detects Reshape-Transpose-Reshape pattern and tries to fuse it into a single ShuffleChannels layer with axis = 1. More...

#include <shuffle_channels_fusion.hpp>

Inheritance diagram for ngraph::pass::ShuffleChannelsFusion:

Public Member Functions

 ShuffleChannelsFusion (const bool reshape_constants_check)
 

Data Fields

 NGRAPH_RTTI_DECLARATION
 

Detailed Description

ShuffleChannelsFusion transformation detects Reshape-Transpose-Reshape pattern and tries to fuse it into a single ShuffleChannels layer with axis = 1.

x' = reshape(x, [N, group, C / group, H, W]) or reshape(x, [N, group, C / group, H * W]) x'' = transpose(x', [0, 2, 1, 3, 4]) or transpose(x', [0, 2, 1, 3]) y = reshape(x'', [N, C, H, W])

Parameters
reshape_constants_checkthe flag that defines the need for additional checks of reshapes constant Additional checks are required when ShuffleChannelsFusion using inside offline transformations and are not necessary when ShuffleChannelsFusion using inside CommonOptimizations

The documentation for this class was generated from the following file: