ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions | Data Fields
luci::sinf::StridedSliceContext Struct Reference
Collaboration diagram for luci::sinf::StridedSliceContext:
Collaboration graph
[legend]

Public Member Functions

 StridedSliceContext (const luci::CircleStridedSlice *node)
 

Data Fields

StridedSliceParams params
 
luci::CircleNode * input = nullptr
 
luci::CircleConst * begin = nullptr
 
luci::CircleConst * end = nullptr
 
luci::CircleConst * strides = nullptr
 
loco::TensorShape effective_input_shape
 
int64_t input_dims = 0
 

Detailed Description

Definition at line 78 of file CircleStridedSlice.cpp.

Constructor & Destructor Documentation

◆ StridedSliceContext()

luci::sinf::StridedSliceContext::StridedSliceContext ( const luci::CircleStridedSlice *  node)
inline

Definition at line 80 of file CircleStridedSlice.cpp.

81 {
82 // check overflow issues
83 assert(static_cast<int16_t>(node->begin_mask()) == node->begin_mask());
84 assert(static_cast<int16_t>(node->ellipsis_mask()) == node->ellipsis_mask());
85 assert(static_cast<int16_t>(node->end_mask()) == node->end_mask());
86 assert(static_cast<int16_t>(node->new_axis_mask()) == node->new_axis_mask());
87 assert(static_cast<int16_t>(node->shrink_axis_mask()) == node->shrink_axis_mask());
88
89 params.begin_mask = node->begin_mask();
90 params.ellipsis_mask = node->ellipsis_mask();
91 params.end_mask = node->end_mask();
92 params.new_axis_mask = node->new_axis_mask();
93 params.shrink_axis_mask = node->shrink_axis_mask();
94
95 input = loco::must_cast<luci::CircleNode *>(node->input());
96 begin = luci::must_cast<luci::CircleConst *>(node->begin());
97 end = luci::must_cast<luci::CircleConst *>(node->end());
98 strides = luci::must_cast<luci::CircleConst *>(node->strides());
99
100 loco::TensorShape input_shape = circle_shape(input);
101 input_dims = static_cast<int64_t>(input_shape.rank());
102 }
luci::CircleStridedSlice::input
loco::Node * input(void) const
Definition CircleStridedSlice.h:35
luci::CircleStridedSlice::begin
loco::Node * begin(void) const
Definition CircleStridedSlice.h:38
luci::CircleStridedSlice::strides
loco::Node * strides(void) const
Definition CircleStridedSlice.h:44
luci::CircleStridedSlice::end
loco::Node * end(void) const
Definition CircleStridedSlice.h:41
loco::must_cast
T must_cast(FeatureEncoder *node)
A helper dynamic_cast that throws when failed.
Definition FeatureCodec.h:80
luci::sinf::circle_shape
loco::TensorShape circle_shape(const luci::CircleNode *node)
Definition CircleShapeInferenceHelper.cpp:143
luci::must_cast
T must_cast(loco::Node *node)
Definition CircleNodeDecl.h:95

References luci::CircleStridedSlice::begin(), begin, luci::CircleStridedSlice::begin_mask(), luci::sinf::StridedSliceParams::begin_mask, luci::sinf::circle_shape(), luci::CircleStridedSlice::ellipsis_mask(), luci::sinf::StridedSliceParams::ellipsis_mask, luci::CircleStridedSlice::end(), end, luci::CircleStridedSlice::end_mask(), luci::sinf::StridedSliceParams::end_mask, luci::CircleStridedSlice::input(), input, input_dims, loco::must_cast(), luci::must_cast(), luci::CircleStridedSlice::new_axis_mask(), luci::sinf::StridedSliceParams::new_axis_mask, params, luci::CircleStridedSlice::shrink_axis_mask(), luci::sinf::StridedSliceParams::shrink_axis_mask, luci::CircleStridedSlice::strides(), and strides.

Field Documentation

◆ begin

luci::CircleConst* luci::sinf::StridedSliceContext::begin = nullptr

Definition at line 105 of file CircleStridedSlice.cpp.

Referenced by StridedSliceContext().

◆ effective_input_shape

loco::TensorShape luci::sinf::StridedSliceContext::effective_input_shape

Definition at line 110 of file CircleStridedSlice.cpp.

◆ end

luci::CircleConst* luci::sinf::StridedSliceContext::end = nullptr

Definition at line 106 of file CircleStridedSlice.cpp.

Referenced by StridedSliceContext().

◆ input

luci::CircleNode* luci::sinf::StridedSliceContext::input = nullptr

Definition at line 104 of file CircleStridedSlice.cpp.

Referenced by StridedSliceContext().

◆ input_dims

int64_t luci::sinf::StridedSliceContext::input_dims = 0

Definition at line 111 of file CircleStridedSlice.cpp.

Referenced by StridedSliceContext().

◆ params

StridedSliceParams luci::sinf::StridedSliceContext::params

Definition at line 103 of file CircleStridedSlice.cpp.

Referenced by StridedSliceContext().

◆ strides

luci::CircleConst* luci::sinf::StridedSliceContext::strides = nullptr

Definition at line 107 of file CircleStridedSlice.cpp.

Referenced by StridedSliceContext().

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