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 ...@@ -40,7 +40,7 @@ namespace ngraph
// //
// A SlicePlan is used to collect parameters for these ops. // A SlicePlan is used to collect parameters for these ops.
// //
struct SlicePlan struct NGRAPH_API SlicePlan
{ {
// Parameters for the Slice // Parameters for the Slice
std::vector<int64_t> begins; std::vector<int64_t> begins;
...@@ -55,7 +55,7 @@ namespace ngraph ...@@ -55,7 +55,7 @@ namespace ngraph
AxisSet reverse_axes; AxisSet reverse_axes;
}; };
SlicePlan make_slice_plan(const Shape& input_shape, SlicePlan NGRAPH_API make_slice_plan(const Shape& input_shape,
const std::vector<int64_t>& begins, const std::vector<int64_t>& begins,
const std::vector<int64_t>& ends, const std::vector<int64_t>& ends,
const std::vector<int64_t>& strides, const std::vector<int64_t>& strides,
......
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