Searched defs:normalized_shape (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_layer_norm_further.py | 27 def __init__(self, begin_norm_axis, begin_params_axis, mul_weight, normalized_shape, argument
51 def __init__(self, begin_norm_axis, begin_params_axis, mul_weight, normalized_shape, argument
75 def __init__(self, begin_norm_axis, begin_params_axis, mul_weight, normalized_shape, argument
99 def __init__(self, begin_norm_axis, begin_params_axis, mul_weight, normalized_shape, argument
121 def __init__(self, begin_norm_axis, begin_params_axis, mul_weight, normalized_shape, argument
|
| D | test_auto_parallel_for_loop_reshape.py | 50 def __init__(self, normalized_shape, eps=1e-5): argument
|
| D | test_auto_parallel_for_loop.py | 50 def __init__(self, normalized_shape, eps=1e-5): argument
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | layer_norm_populate_v0.cc | 32 auto normalized_shape = layer_norm_prim->normalizedShape(); in PopulateLayerNormParameterV0() local
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_layernorm.py | 38 def pt_me_layernorm(input_data, normalized_shape, gamma, beta, axis): argument
|
| D | test_layernorm_grad.py | 48 def py_me_layernorm_grad(input_data, normalized_shape, gamma, beta, axis, gradients): argument
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | layers.py | 204 def __init__(self, normalized_shape, eps=1e-5, param_init_type=mstype.float32): argument
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | normalization.py | 883 normalized_shape, argument
|