Class EmbeddingLookupFusion
Defined in File embedding_lookup_fusion.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class EmbeddingLookupFusion : public mindspore::ops::BaseOperator
EmbeddingLookupFusion defined EmbeddingLookup operator prototype of lite.
Public Functions
-
inline EmbeddingLookupFusion()
Constructor.
-
void Init(const float max_norm = 0.0)
Method to init the op’s attributes.
- Parameters
max_norm – [in] Define the max l2-norm value of each embedding. Each embedding will be clip if l2-norm is larger than this value.
-
void set_max_norm(const float max_norm)
Method to set max_norm attribute.
- Parameters
max_norm – [in] Define the max l2-norm value of each embedding. Each embedding will be clip if l2-norm is larger than this value.
-
float get_max_norm() const
Method to get max_norm attribute.
- Returns
a value which indicates a max l2-norm value.
-
inline EmbeddingLookupFusion()