| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | bprop_util.py | 30 def __init__(self, func, wrt_params, params, grad_op, sens): argument
33 self.wrt_params = wrt_params
35 if self.wrt_params and params:
45 if self.wrt_params:
82 wrt_params = False
84 wrt_params = True
88 grad_op = GradOperation(get_all=wrt_inputs, get_by_list=wrt_params, sens_param=with_sens_param)
89 grad = Bprop(func, wrt_params, params, grad_op, grads_wrt_outputs)
|
| /third_party/mindspore/tests/st/high_grad/ |
| D | test_highgrad_train.py | 30 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
35 self.wrt_params = wrt_params
37 if self.wrt_params:
42 if self.wrt_params:
48 if self.wrt_params:
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_graph_param_cases.py | 32 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
37 self.wrt_params = wrt_params
39 if self.wrt_params:
43 if self.wrt_params:
|
| D | test_multi_grad.py | 30 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
35 self.wrt_params = wrt_params
37 if self.wrt_params:
41 if self.wrt_params:
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_graph_param_transform.py | 30 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
35 self.wrt_params = wrt_params
37 if self.wrt_params:
41 if self.wrt_params:
|
| D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 27 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
32 self.wrt_params = wrt_params
34 if self.wrt_params:
38 if self.wrt_params:
53 network=network, wrt_params=True, real_inputs_count=real_inputs_count)
|
| D | test_pynative_hook_grad.py | 68 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
73 self.wrt_params = wrt_params
75 if self.wrt_params:
79 if self.wrt_params:
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_auto_monad_gpu.py | 37 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
42 self.wrt_params = wrt_params
44 if self.wrt_params:
49 if self.wrt_params:
55 if self.wrt_params:
77 network=network, wrt_params=True, real_inputs_count=real_inputs_count)
|
| D | test_auto_monad_mindtester.py | 33 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
38 self.wrt_params = wrt_params
40 if self.wrt_params:
45 if self.wrt_params:
51 if self.wrt_params:
83 network=network, wrt_params=True, real_inputs_count=real_inputs_count)
|
| /third_party/mindspore/tests/ut/python/parameter_feature/ |
| D | test_var_grad.py | 77 def __init__(self, func, wrt_params, params, grad_op, sens=None): argument
80 self.wrt_params = wrt_params
82 if self.wrt_params and params:
93 if self.wrt_params:
|