Searched defs:max_norm (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | embedding_lookup_fusion.cc | 23 void EmbeddingLookupFusion::set_max_norm(const float max_norm) { (void)this->AddAttr(kMaxNorm, Make… in set_max_norm()
29 void EmbeddingLookupFusion::Init(const float max_norm) { this->set_max_norm(max_norm); } in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | embedding_lookup_fp32.c | 21 void l2_regulate(float *data, int size, float max_norm) { in l2_regulate()
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 225 max_norm=None, sparse=True, vocab_cache_size=0): argument
477 … slice_mode='batch_slice', feature_num_list=None, max_norm=None, sparse=True, operator='SUM'): argument
|
| D | thor_layer.py | 722 max_norm=None, sparse=True, vocab_cache_size=0): argument
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 318 def construct(self, input_x, max_norm): argument
353 def __init__(self, vocab_size, embedding_size, max_norm=None): argument
|