Home
last modified time | relevance | path

Searched defs:input_mean (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_batchnorm.py25 def __init__(self, input_scale, input_bias, input_mean, input_variance, is_training): argument
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_layer_norm_x_backprop_v2.cc39 …auto input_mean = TensorConstructUtils::CreateOnesTensor(kFloat16, std::vector<int64_t>{1, 128, 1}… in TEST_F() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dlayernorm_grad.cc81 auto input_mean = in_tensors_.at(3); in Execute() local
Dbn_grad.cc79 auto *input_mean = in_tensors_.at(3); in Execute() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dlayernorm_fp16_grad.cc90 auto input_mean = in_tensors_.at(kNumInputDim_3); in DoExecute() local
Dbn_fp16_grad.cc79 auto *input_mean = in_tensors_.at(kNumInputDim_3); in DoExecute() local