| /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
38 self.real_inputs_count = real_inputs_count
44 if self.real_inputs_count is None or self.sens_param is False:
46 real_inputs = inputs[:self.real_inputs_count]
47 sense_param_inputs = inputs[self.real_inputs_count:]
50 if self.real_inputs_count is None or self.sens_param is False:
52 real_inputs = inputs[:self.real_inputs_count]
53 sense_param_inputs = inputs[self.real_inputs_count:]
62 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
64 network=network, real_inputs_count=real_inputs_count)
[all …]
|
| D | test_multi_grad.py | 30 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
36 self.real_inputs_count = real_inputs_count
42 if self.real_inputs_count is None or self.sens_param is False:
44 real_inputs = inputs[:self.real_inputs_count]
45 sense_param_inputs = inputs[self.real_inputs_count:]
48 if self.real_inputs_count is None or self.sens_param is False:
50 real_inputs = inputs[:self.real_inputs_count]
51 sense_param_inputs = inputs[self.real_inputs_count:]
60 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
62 network=network, real_inputs_count=real_inputs_count)
[all …]
|
| /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
36 self.real_inputs_count = real_inputs_count
42 if self.real_inputs_count is None or self.sens_param is False:
44 real_inputs = inputs[:self.real_inputs_count]
45 sense_param_inputs = inputs[self.real_inputs_count:]
48 if self.real_inputs_count is None or self.sens_param is False:
50 real_inputs = inputs[:self.real_inputs_count]
51 sense_param_inputs = inputs[self.real_inputs_count:]
60 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
62 network=network, real_inputs_count=real_inputs_count)
[all …]
|
| D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 27 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
33 self.real_inputs_count = real_inputs_count
39 if self.real_inputs_count is None or self.sens_param is False:
41 real_inputs = inputs[:self.real_inputs_count]
42 sense_param_inputs = inputs[self.real_inputs_count:]
44 if self.real_inputs_count is None or self.sens_param is False:
46 real_inputs = inputs[:self.real_inputs_count]
47 sense_param_inputs = inputs[self.real_inputs_count:]
51 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
53 network=network, wrt_params=True, real_inputs_count=real_inputs_count)
[all …]
|
| D | test_pynative_hook_grad.py | 68 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
74 self.real_inputs_count = real_inputs_count
80 if self.real_inputs_count is None or self.sens_param is False:
82 real_inputs = inputs[:self.real_inputs_count]
83 sense_param_inputs = inputs[self.real_inputs_count:]
85 if self.real_inputs_count is None or self.sens_param is False:
87 real_inputs = inputs[:self.real_inputs_count]
88 sense_param_inputs = inputs[self.real_inputs_count:]
92 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
94 network=network, real_inputs_count=real_inputs_count)
|
| /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
36 self.real_inputs_count = real_inputs_count
41 if self.real_inputs_count is None or self.sens_param is False:
46 real_inputs = inputs[:self.real_inputs_count]
47 sense_param_inputs = inputs[self.real_inputs_count:]
58 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
60 network=network, real_inputs_count=real_inputs_count)
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_auto_monad_mindtester.py | 33 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
39 self.real_inputs_count = real_inputs_count
44 if self.real_inputs_count is None or self.sens_param is False:
49 real_inputs = inputs[:self.real_inputs_count]
50 sense_param_inputs = inputs[self.real_inputs_count:]
61 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
63 network=network, real_inputs_count=real_inputs_count)
71 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
73 network=network, real_inputs_count=real_inputs_count)
81 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
[all …]
|
| D | test_auto_monad_gpu.py | 37 def __init__(self, grad, network, wrt_params=False, real_inputs_count=None): argument
43 self.real_inputs_count = real_inputs_count
48 if self.real_inputs_count is None or self.sens_param is False:
53 real_inputs = inputs[:self.real_inputs_count]
54 sense_param_inputs = inputs[self.real_inputs_count:]
65 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
67 network=network, real_inputs_count=real_inputs_count)
75 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
77 network=network, wrt_params=True, real_inputs_count=real_inputs_count)
|