Public Member Functions
ngraph::pass::ConvertBatchToSpace Class Reference

ConvertBatchToSpace transformation decomposes BatchToSpace layer to Reshape->Transpose->Reshape->Crop. More...

#include <convert_batch_to_space.hpp>

Inheritance diagram for ngraph::pass::ConvertBatchToSpace:

Public Member Functions

 ConvertBatchToSpace (bool convert_by_elements=true)
 

Detailed Description

ConvertBatchToSpace transformation decomposes BatchToSpace layer to Reshape->Transpose->Reshape->Crop.

Parameters
convert_by_elements- reduces the maximum number of dimensions that arise during the transformation if enabled. Default value: true. false - BatchToSpace decomposes to Reshape->Transpose->Reshape->Crop. During transformation, the number of tensor dimensions can be increased by length of block_shape input of BatchToSpace layer. true - BatchToSpace decomposes to N x (Reshape->Transpose->Reshape)->Crop, where N = length of block_shape input of BatchToSpace layer. During transformation, the number of tensor dimensions can be increased by 1.

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