Searched refs:padding_idx (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 99 dtype=mstype.float32, padding_idx=None): argument
109 self.padding_idx = padding_idx
110 if padding_idx is not None:
111 self.padding_idx = validator.check_int_range(padding_idx, 0, vocab_size, Rel.INC_BOTH,
116 self.init_tensor[self.padding_idx] = 0
146 …ab_size, self.embedding_size, self.use_one_hot, self.embedding_table, self.dtype, self.padding_idx)
|
| D | thor_layer.py | 571 dtype=mstype.float32, padding_idx=None): argument
581 self.padding_idx = padding_idx
582 if padding_idx is not None:
583 self.padding_idx = Validator.check_int_range(padding_idx, 0, vocab_size, Rel.INC_BOTH,
586 self.init_tensor[self.padding_idx] = 0
648 …ab_size, self.embedding_size, self.use_one_hot, self.embedding_table, self.dtype, self.padding_idx)
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 340 def __init__(self, vocab_size, embedding_size, padding_idx=None): argument
343 padding_idx=padding_idx)
1381 'block': Embedding(vocab_size=10, embedding_size=3, padding_idx=2),
|
| /third_party/mbedtls/library/ |
| D | ssl_msg.c | 1380 size_t const padding_idx = rec->data_len - padlen; in mbedtls_ssl_decrypt_buf() local
1390 const size_t mask = mbedtls_ct_size_mask_ge( idx, padding_idx ); in mbedtls_ssl_decrypt_buf()
|