Class Splice
Defined in File splice.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class Splice : public mindspore::ops::BaseOperator
All defined All operator prototype of lite.
Public Functions
-
inline Splice()
Constructor.
-
void Init(const std::vector<int64_t> &contexts, const std::vector<int64_t> &forward_indexes, int64_t output_dims)
Method to init the op’s attributes.
- Parameters
contexts – [in] Define the contexts.
forward_indexes – [in] Define the forward indexes.
output_dims – [in] Define the output dims.
-
void set_context(const std::vector<int64_t> &contexts)
Method to set contexts attributes.
- Parameters
contexts – [in] Define the contexts.
-
void set_forward_indexes(const std::vector<int64_t> &forward_indexes)
Method to set forward_indexes attributes.
- Parameters
forward_indexes – [in] Define the forward_indexes.
-
void set_output_dim(int64_t output_dim)
Method to set output_dim attributes.
- Parameters
output_dim – [in] Define the output_dim.
-
std::vector<int64_t> get_context() const
Method to set context attributes.
- Parameters
context – [in] Define the context.
-
std::vector<int64_t> get_forward_indexes() const
Method to set forward_indexes attributes.
- Parameters
forward_indexes – [in] Define the forward_indexes.
-
int64_t get_output_dim() const
Method to set output_dim attributes.
- Parameters
output_dim – [in] Define the output_dim.
-
inline Splice()