Searched refs:load_torch_embedding_weights (Results 1 – 2 of 2) sorted by relevance
35 from .torch import dump_torch_embedding_weights, load_torch_embedding_weights
380 def load_torch_embedding_weights(where, emb): function429 load_torch_embedding_weights(where, module)