convert_batch_to_space.hpp

// Copyright (C) 2018-2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
 
#pragma once
 
#include <vector>
#include <memory>
 
#include <transformations_visibility.hpp>
 
#include <ngraph/ops.hpp>
#include <ngraph/pass/graph_rewrite.hpp>
 
namespace ngraph {
namespace pass {
 
class TRANSFORMATIONS_API ConvertBatchToSpace;
 
}  // namespace pass
}  // namespace ngraph
 
/**
 * @ingroup ie_transformation_common_api
 * @brief ConvertBatchToSpace transformation decomposes BatchToSpace layer to Reshape->Transpose->Reshape->Crop.
 *
 * @param 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.
 *
 */
 
class ngraph::pass::ConvertBatchToSpace: public ngraph::pass::MatcherPass {
public:
    NGRAPH_RTTI_DECLARATION;
    explicit ConvertBatchToSpace(bool convert_by_elements = true) : MatcherPass() {
        if (convert_by_elements)
            convert_batch_to_space_by_elements();
        else
            convert_batch_to_space();
    }
 
private:
    void convert_batch_to_space();
    void convert_batch_to_space_by_elements();
};