Searched defs:beta_init (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | normalization.py | 61 beta_init='zeros', argument
446 beta_init='zeros', argument
598 beta_init='zeros', argument
742 beta_init='zeros', argument
786 beta_init='zeros', argument
1136 …nit__(self, num_groups, num_channels, eps=1e-05, affine=True, gamma_init='ones', beta_init='zeros', argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_instance_norm_nd.py | 71 def __init__(self, n, gamma_init=0.5, beta_init=0.5): argument
81 def __init__(self, n, gamma_init=0.5, beta_init=0.5): argument
91 def __init__(self, n, gamma_init=0.5, beta_init=0.5): argument
|
| D | test_batchnorm_op.py | 209 def __init__(self, affine=True, gamma_init='ones', beta_init='zeros', moving_mean_init='zeros', argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/extend/layer/ |
| D | normalization.py | 81 beta_init='zeros', argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/pynative/ |
| D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 93 …def __init__(self, input_shape, norm_shape, gamma_shape, beta_shape, gamma_init=None, beta_init=No… argument
|