Returns the op that generates these parameters (to be used for colocating other ops with it).
Returns the op that generates these parameters (to be used for colocating other ops with it).
Returns the dynamic shape of this parameters tensor.
Returns the dynamic shape of this parameters tensor.
Gathers the embeddings corresponding to indices
from parameters
.
Returns the static shape of this parameters tensor.
Returns the static shape of this parameters tensor.
Trait for specifying supported embedding parameter types.