slice_plan.hpp

// Copyright (C) 2018-2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
 
#pragma once
 
#include <set>
 
#include "ngraph/axis_set.hpp"
#include "ngraph/shape.hpp"
 
namespace ngraph
{
    //
    // In various places, like ConstantFolding and DynElimination, it is
    // useful to transform DynSlice by converting it to a sequence of ops:
    //
    //      Slice    (to do the basic slicing)
    //        |
    //        v
    //     Reshape   (non-transposing, to handle shrinks)
    //        |
    //        v
    //     Reverse   (to emulate backwards stride)
    //
    // (The Reshape, Reverse, or both may be omitted if they would just be
    // identities.)
    //
    // A SlicePlan is used to collect parameters for these ops.
    //
    struct NGRAPH_API SlicePlan
    {
        // Parameters for the Slice
        std::vector<int64_t> begins;
        std::vector<int64_t> ends;
        std::vector<int64_t> strides;
 
        // Shapes coming into, and going out of, the Reshape.
        Shape reshape_in_shape;
        Shape reshape_out_shape;
 
        // Parameters for the Reverse
        AxisSet reverse_axes;
 
        bool operator==(const SlicePlan& other) const;
        bool operator!=(const SlicePlan& other) const;
    };
 
    SlicePlan NGRAPH_API make_slice_plan(const Shape& input_shape,
                                         const std::vector<int64_t>& begins,
                                         const std::vector<int64_t>& ends,
                                         const std::vector<int64_t>& strides,
                                         const AxisSet& lower_bounds_mask,
                                         const AxisSet& upper_bounds_mask,
                                         const AxisSet& new_axis_mask,
                                         const AxisSet& shrink_axis_mask,
                                         const AxisSet& ellipsis_mask);
} // namespace ngraph