Searched defs:moving_mean (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_batchnorm_op.py | 27 def __init__(self, c, weight, bias, moving_mean, moving_var_init): argument
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_batchnorm.py | 36 def get_output(x, weight, bias, moving_mean, moving_var, is_training, enable_graph_kernel=False): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | batch_norm_cpu_kernel.cc | 96 auto moving_mean = reinterpret_cast<float *>(inputs[3]->addr); in Launch() local
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_batchnorm_op.py | 28 def __init__(self, c, weight, bias, moving_mean, moving_var_init, use_batch_statistics=None): argument
|