Searched refs:use_one_hot (Results 1 – 4 of 4) sorted by relevance
| /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
104 validator.check_value_type('use_one_hot', use_one_hot, [bool], self.cls_name)
106 self.use_one_hot = use_one_hot
135 if self.use_one_hot:
146 …self.vocab_size, self.embedding_size, self.use_one_hot, self.embedding_table, self.dtype, self.pad…
|
| D | thor_layer.py | 570 def __init__(self, vocab_size, embedding_size, use_one_hot=False, embedding_table='normal', argument
576 Validator.check_value_type('use_one_hot', use_one_hot, [bool], self.cls_name)
578 self.use_one_hot = use_one_hot
630 if self.use_one_hot:
648 …self.vocab_size, self.embedding_size, self.use_one_hot, self.embedding_table, self.dtype, self.pad…
|
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | convert_utils.py | 76 use_one_hot=False)
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_lstm.py | 70 …self.embedding = nn.Embedding(vocab_size, embed_size, use_one_hot=False, embedding_table=Tensor(we…
|