Class EmbeddingLookupFusion

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.