Searched defs:input_dy (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_batchnorm_grad.py | 28 …def construct(self, input_dy, input_x, input_scale, input_save_mean, input_save_inv_variance, inpu… argument
33 def get_output(input_dy, input_x, input_scale, input_save_mean, input_save_inv_variance, input_reve… argument
|
| D | test_gelu.py | 69 def gelu_grad(input_dy, input_x, input_y, enable_graph_kernel=False): argument
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_layer_norm_beta_gamma_backprop_v2.cc | 40 auto input_dy = TensorConstructUtils::CreateOnesTensor(kFloat16, std::vector<int64_t>{1, 128, 1}); in TEST_F() local
|
| D | test_ops_layer_norm_x_backprop_v2.cc | 37 auto input_dy = 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/ |
| D | layernorm_grad.cc | 79 auto input_dy = in_tensors_.at(1); in Execute() local
|
| D | deconvolution_grad_filter.cc | 75 auto *input_dy = in_tensors_.at(0); in Execute() local
|
| D | convolution_grad_input.cc | 81 auto *input_dy = in_tensors_.at(0); in Execute() local
|
| D | convolution_grad_filter.cc | 88 auto *input_dy = in_tensors_.at(0); in Execute() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | layernorm_fp16_grad.cc | 88 auto input_dy = in_tensors_.at(kNumInputDim_1); in DoExecute() local
|
| D | convolution_fp16_grad_input.cc | 80 auto *input_dy = in_tensors_.at(0); in DoExecute() local
|
| D | convolution_fp16_grad_filter.cc | 85 auto *input_dy = in_tensors_.at(0); in DoExecute() local
|