Class SliceFusion
Defined in File slice_fusion.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class SliceFusion : public mindspore::ops::BaseOperator
SliceFusion defined Slice operator prototype of lite.
Public Functions
-
inline SliceFusion()
Constructor.
-
void Init(const std::vector<int64_t> &axes)
Method to init the op’s attributes.
- 参数
axes – [in] Define this operation will be performed on which axes.
-
void set_axes(const std::vector<int64_t> &axes)
Method to set axes attribute.
- 参数
axes – [in] Define this operation will be performed on which axes.
-
std::vector<int64_t> get_axes() const
Method to get axes attribute.
- 返回
axes.
-
inline SliceFusion()