Searched defs:input_mean (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_parameterized_truncated_normal.py | 27 … def __init__(self, input_shape, input_mean, input_stdevs, input_min, input_max, seed=0, seed2=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/ |
| D | batchnorm.cc | 60 const auto &input_mean = inputs[3]; in Train() local
148 const auto &input_mean = inputs[3]; in Expand() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/ |
| D | test_batchnorm.py | 25 def __init__(self, input_scale, input_bias, input_mean, input_variance, is_training): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | layer_norm_grad_grad_cpu_kernel.cc | 77 DATA_T *input_x, DATA_T *input_mean, DATA_T *x_hat, DATA_T *dy_gamma) { in shard_inner_mean()
100 DATA_T *input_mean) { in shard_outer_mean()
184 auto input_mean = static_cast<DATA_T *>(inputs[3]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | log_normal_reverse.cc | 55 float input_mean = 1.0; in DoCompute() local
|
| D | layer_norm_grad_grad.cc | 74 … const size_t &mean_num, T *input_x, T *input_dy, T *input_mean, T *input_gamma, in SwitchParallelCompute0()
100 … const size_t &mean_num, T *input_x, T *input_dy, T *input_mean, T *input_gamma, in SwitchParallelCompute1()
171 auto input_mean = reinterpret_cast<T *>(ctx.Input(3)->GetData()); in LayerNormGradGradCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/ |
| D | layernorm_grad.cc | 81 auto input_mean = in_tensors_.at(3); in DoExecute() local
|
| D | bn_grad.cc | 64 auto *input_mean = in_tensors_.at(3); in DoExecute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16_grad/ |
| D | layernorm_fp16_grad.cc | 90 auto input_mean = in_tensors_.at(kNumInputDim_3); in DoExecute() local
|
| D | bn_fp16_grad.cc | 79 auto *input_mean = in_tensors_.at(kNumInputDim_3); in DoExecute() local
|