Loading...
Searching...
No Matches
embedding_lookup_operator.h File Reference

Go to the source code of this file.

Classes

struct  opennn::EmbeddingLookupOperator
 

Namespaces

namespace  opennn
 

Functions

void opennn::embedding_lookup_forward (const TensorView &, const TensorView &, const TensorView &, TensorView &, Index, Index, Index, bool scale_embedding, bool add_positional_encoding, const TensorView &weight_scale={})
 
void opennn::embedding_lookup_backward (const TensorView &, const TensorView &, const TensorView &, const TensorView &, Index, Index, Index, bool scale_embedding)
 
void opennn::compute_token_valid_lengths (const TensorView &, Index, vector< Index > &)