Searched defs:beta_init (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | normalization.py | 53 beta_init='zeros', argument
356 beta_init='zeros', argument
464 beta_init='zeros', argument
566 beta_init='zeros', argument
684 beta_init='zeros', argument
805 beta_init='zeros', argument
887 beta_init='zeros', argument
994 beta_init='zeros'): argument
1105 …it__(self, num_groups, num_channels, eps=1e-05, affine=True, gamma_init='ones', beta_init='zeros'): argument
|
| D | quant.py | 679 beta_init='zeros', argument
919 beta_init='zeros', argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_batchnorm_op.py | 208 def __init__(self, affine=True, gamma_init='ones', beta_init='zeros', moving_mean_init='zeros', argument
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 94 …def __init__(self, input_shape, norm_shape, gamma_shape, beta_shape, gamma_init=None, beta_init=No… argument
|