| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 98 def __init__(self, vocab_size, embedding_size, use_one_hot=False, embedding_table='normal', argument
223 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
371 def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size): argument
476 def __init__(self, vocab_size, embedding_size, field_size, param_init='normal', target='CPU', argument
|
| D | thor_layer.py | 570 def __init__(self, vocab_size, embedding_size, use_one_hot=False, embedding_table='normal', argument
720 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
899 def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/text/ |
| D | bindings.cc | 56 … [](const py::list &paths, const int32_t vocab_size, const float character_coverage, in __anonb29b31e90502()
57 const SentencePieceModel model_type, const py::dict ¶ms) { in __anonb29b31e90502()
|
| /third_party/mindspore/mindspore/dataset/text/ |
| D | utils.py | 104 …def from_file(cls, file_path, delimiter="", vocab_size=None, special_tokens=None, special_first=Tr… argument
158 def from_dataset(cls, dataset, col_names, vocab_size, character_coverage, model_type, params): argument
197 def from_file(cls, file_path, vocab_size, character_coverage, model_type, params): argument
|
| /third_party/mindspore/tests/st/dynamic_shape/ |
| D | test_dynamic_shape_embedding.py | 27 def __init__(self, vocab_size, embedding_size, target="CPU"): argument
|
| /third_party/mindspore/tests/st/ps/part_ps/ |
| D | test_ps_embedding_heterogeneous_conv2d_adam.py | 51 def __init__(self, in_channels, out_channels, kernel_size, vocab_size, embedding_size, argument
126 kernel_size=5, vocab_size=5, embedding_size=1, output_channels=3072, argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/ |
| D | vocab.cc | 135 … Vocab::BuildFromFileCpp(const std::string &path, const std::string &delimiter, int32_t vocab_size, in BuildFromFileCpp()
197 …tus Vocab::BuildFromFile(const std::string &path, const std::string &delimiter, int32_t vocab_size, in BuildFromFile()
|
| D | sentence_piece_vocab.cc | 33 …tencePieceVocab::BuildFromFile(const std::vector<std::string> &path_list, const int32_t vocab_size, in BuildFromFile()
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_lstm.py | 62 def __init__(self, vocab_size, embed_size, num_hiddens, num_layers, argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | build_sentence_piece_vocab_node.cc | 32 … const std::vector<std::string> &col_names, int32_t vocab_size, in BuildSentenceVocabNode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | build_sentence_piece_vocab_op.cc | 25 … std::vector<std::string> col_names, int32_t vocab_size, in BuildSentencePieceVocabOp()
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_vocab.py | 152 def test_config(lookup_str, vocab_size, special_tokens, special_first): argument
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _ps_context.py | 201 def _insert_hash_table_size(name, cache_vocab_size, embedding_size, vocab_size): argument
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | multifieldembeddinglookup_parallel.py | 178 def __init__(self, vocab_size, embedding_size, field_size, argument
202 def __init__(self, vocab_size, embedding_size, field_size, argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | dataset.py | 151 …def __init__(self, batch_size, max_seq_length, vocab_size, keep_first_unchange=True, keep_last_unc… argument
|
| D | cell_wrapper.py | 177 def __init__(self, network, vocab_size=21128): argument
|
| D | model.py | 60 vocab_size=21128, argument
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 65 vocab_size=32000, argument
116 vocab_size, argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 247 const std::vector<std::string> &col_names, int32_t vocab_size, float character_coverage, in BuildSentencePieceVocab()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 146 const std::unordered_map<std::string, std::string> ¶ms) { in __anon4440814b0c02()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.cc | 26 size_t vocab_size) { in InsertHashTableSize()
1125 size_t vocab_size = item.second.vocab_size; in DumpHashTables() local
|
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_row_tensor.py | 439 def __init__(self, vocab_size, embedding_size, target='CPU'): argument
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 519 … def __init__(self, vocab_size, embedding_size, parallel_config=default_embedding_parallel_config, argument
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 340 def __init__(self, vocab_size, embedding_size, padding_idx=None): argument
353 def __init__(self, vocab_size, embedding_size, max_norm=None): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 553 const std::vector<std::vector<char>> &col_names, int32_t vocab_size, float character_coverage, in BuildSentencePieceVocabCharIF()
|