Searched refs:dump_torch_embedding_weights (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/libopus/dnn/torch/weight-exchange/wexchange/torch/ |
| H A D | __init__.py | 35 from .torch import dump_torch_embedding_weights, load_torch_embedding_weights
|
| H A D | torch.py | 365 def dump_torch_embedding_weights(where, embed, name='embed', scale=1/128, sparse=False, diagonal=Fa… function
402 return dump_torch_embedding_weights(where, module, name, **kwargs)
|
| /aosp_15_r20/external/libopus/dnn/torch/fargan/ |
| H A D | dump_fargan_weights.py | 106 …wexchange.torch.dump_torch_embedding_weights(writer, module, name.replace('.', '_'), quantize=quan…
|