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>
Public Member Functions | |
| ShuffleChannelsFusion (const bool reshape_constants_check) | |
Data Fields | |
| NGRAPH_RTTI_DECLARATION | |
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])
| reshape_constants_check | the 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 |