Unverified Commit 69c5457d authored by Evgenya Stepyreva's avatar Evgenya Stepyreva Committed by GitHub

Expose slice plan (#4526)

* Expose slice plan

* Style
Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
parent 1836001d
Pipeline #347 failed with stages
......@@ -40,7 +40,7 @@ namespace ngraph
//
// A SlicePlan is used to collect parameters for these ops.
//
struct SlicePlan
struct NGRAPH_API SlicePlan
{
// Parameters for the Slice
std::vector<int64_t> begins;
......@@ -55,13 +55,13 @@ namespace ngraph
AxisSet reverse_axes;
};
SlicePlan 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);
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);
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment