Home
last modified time | relevance | path

Searched refs:sens_param (Results 1 – 25 of 168) sorted by relevance

1234567

/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_graph_param_cases.py36 self.sens_param = self.grad.sens_param
44 if self.real_inputs_count is None or self.sens_param is False:
50 if self.real_inputs_count is None or self.sens_param is False:
62 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
63 super().__init__(grad=C.GradOperation(sens_param=sens_param),
72 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
73 super().__init__(grad=C.GradOperation(get_all=True, sens_param=sens_param),
Dtest_multi_grad.py34 self.sens_param = self.grad.sens_param
42 if self.real_inputs_count is None or self.sens_param is False:
48 if self.real_inputs_count is None or self.sens_param is False:
60 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
61 super().__init__(grad=C.GradOperation(sens_param=sens_param),
70 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
71 super().__init__(grad=C.GradOperation(get_all=True, sens_param=sens_param),
/third_party/mindspore/tests/st/high_grad/
Dtest_highgrad_train.py34 self.sens_param = self.grad.sens_param
41 if self.real_inputs_count is None or self.sens_param is False:
58 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
59 super().__init__(grad=GradOperation(sens_param=sens_param),
104 grad_net = GradOfFirstInput(net, sens_param=False)
Dtest_highgrad_param.py42 self.grad = ops.GradOperation(get_by_list=True, sens_param=False)
54 self.grad = ops.GradOperation(get_all=True, sens_param=False)
/third_party/mindspore/tests/st/pynative/
Dtest_graph_param_transform.py34 self.sens_param = self.grad.sens_param
42 if self.real_inputs_count is None or self.sens_param is False:
48 if self.real_inputs_count is None or self.sens_param is False:
60 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
61 super().__init__(grad=C.GradOperation(sens_param=sens_param),
70 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
71 super().__init__(grad=C.GradOperation(get_all=True, sens_param=sens_param),
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py31 self.sens_param = self.grad.sens_param
39 if self.real_inputs_count is None or self.sens_param is False:
44 if self.real_inputs_count is None or self.sens_param is False:
51 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
52 super().__init__(grad=GradOperation(get_all=True, get_by_list=True, sens_param=sens_param),
157 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
158 super().__init__(grad=GradOperation(sens_param=sens_param),
Dtest_pynative_hook_grad.py72 self.sens_param = self.grad.sens_param
80 if self.real_inputs_count is None or self.sens_param is False:
85 if self.real_inputs_count is None or self.sens_param is False:
92 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
93 super().__init__(grad=GradOperation(get_all=True, sens_param=sens_param),
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dgrad_var_prepare.cc33 … const AnfNodePtr &func_node, bool is_unpack, bool sens_param) { in GenerateUnpackGraphNode() argument
40 …auto unpack_graph = std::make_shared<prim::UnpackGraphPrimitive>("unpack_graph", sens_param, true); in GenerateUnpackGraphNode()
49 …auto unpack_graph = std::make_shared<prim::UnpackGraphPrimitive>("unpack_graph", sens_param, false… in GenerateUnpackGraphNode()
134 const bool sens_param = grad_op_ptr->sens_param(); in operator ()() local
135 inputs_x[1] = GenerateUnpackGraphNode(node, inputs_y, func_node, is_unpack, sens_param); in operator ()()
/third_party/mindspore/mindspore/nn/grad/
Dcell_grad.py27 self.first_grad_op = C.GradOperation(sens_param=True, get_all=True)
37 self.first_grad_single_value_op = C.GradOperation(sens_param=True)
79 self.second_grad_op = C.GradOperation(sens_param=True)
139 self.grad = C.GradOperation(get_all=True, sens_param=True)
140 self.grad_single_value = C.GradOperation(sens_param=True)
/third_party/mindspore/tests/st/auto_monad/
Dtest_auto_monad_mindtester.py37 self.sens_param = self.grad.sens_param
44 if self.real_inputs_count is None or self.sens_param is False:
61 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
62 super().__init__(grad=GradOperation(sens_param=sens_param),
71 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
72 super().__init__(grad=GradOperation(get_all=True, sens_param=sens_param),
81 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
82 super().__init__(grad=GradOperation(get_all=True, get_by_list=True, sens_param=sens_param),
199 grad_net = GradOfAllInputs(net, sens_param=False)
223 self.grad = GradOfAllInputs(self.f, sens_param=False)
[all …]
Dtest_auto_monad_gpu.py41 self.sens_param = self.grad.sens_param
48 if self.real_inputs_count is None or self.sens_param is False:
65 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
66 super().__init__(grad=GradOperation(get_all=True, sens_param=sens_param),
75 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
76 super().__init__(grad=GradOperation(get_all=True, get_by_list=True, sens_param=sens_param),
/third_party/mindspore/mindspore/ops/composite/
Dbase.py319 def __init__(self, get_all=False, get_by_list=False, sens_param=False): argument
326 if not isinstance(sens_param, bool):
331 self.sens_param = sens_param
332 GradOperation_.__init__(self, 'grad', get_all, get_by_list, sens_param)
340 if self.sens_param:
362 grad_ = GradOperation(self.get_all, self.get_by_list, self.sens_param)
/third_party/mindspore/mindspore/nn/wrap/
Dcell_wrapper.py171 self.grad = C.GradOperation(get_by_list=True, sens_param=(sens is not None))
250 def __init__(self, network, weights=None, get_all=False, get_by_list=False, sens_param=False): argument
268 self.sens_param = sens_param
269 …d = C.GradOperation(get_all=self.get_all, get_by_list=self.get_by_list, sens_param=self.sens_param)
273 if self.sens_param:
341 self.grad = C.GradOperation(get_by_list=True, sens_param=True)
/third_party/mindspore/tests/st/ops/ascend/
Dtest_lstm_op.py29 def __init__(self, network, sens_param): argument
31 self.grad = c.GradOperation(get_all=True, get_by_list=True, sens_param=sens_param)
156 grad_net_inp = GradOfAllInputsAndParams(net, sens_param=False)
170 grad_net_inp_pynative = GradOfAllInputsAndParams(net_pynative, sens_param=False)
Dtest_rnn_op.py28 def __init__(self, network, sens_param): argument
30 self.grad = c.GradOperation(get_all=True, get_by_list=True, sens_param=sens_param)
159 grad_net_inp = GradOfAllInputsAndParams(net, sens_param=False)
174 grad_net_inp_pynative = GradOfAllInputsAndParams(net_pynative, sens_param=False)
Dtest_gru_op.py28 def __init__(self, network, sens_param): argument
30 self.grad = c.GradOperation(get_all=True, get_by_list=True, sens_param=sens_param)
159 grad_net_inp = GradOfAllInputsAndParams(net, sens_param=False)
174 grad_net_inp_pynative = GradOfAllInputsAndParams(net_pynative, sens_param=False)
/third_party/mindspore/tests/st/ops/gpu/
Dtest_gather_grad_op.py49 grad_net = GradOperation(get_all=True, sens_param=True)(net)
68 grad_net = GradOperation(get_all=True, sens_param=True)(net)
87 grad_net = GradOperation(get_all=True, sens_param=True)(net)
106 grad_net = GradOperation(get_all=True, sens_param=True)(net)
/third_party/mindspore/tests/st/pynative/ms_function/
Dtest_pynative_ms_function.py49 grad = P.GradOperation(get_all=True, get_by_list=True, sens_param=False)
82 grad = P.GradOperation(get_all=True, get_by_list=True, sens_param=False)
139 grad = P.GradOperation(get_all=True, get_by_list=True, sens_param=False)
194 grad = P.GradOperation(get_all=True, get_by_list=True, sens_param=False)
254 grad = P.GradOperation(get_all=True, get_by_list=True, sens_param=False)
/third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/
Dtest_gelu_grad_sens.py30 self.grad = GradOperation(get_all=True, sens_param=True)
74 self.grad = GradOperation(get_all=True, sens_param=True)
Dtest_fast_gelu_grad_sens.py30 self.grad = GradOperation(get_all=True, sens_param=True)
74 self.grad = GradOperation(get_all=True, sens_param=True)
/third_party/mindspore/tests/ut/python/parameter_feature/
Dtest_var_grad.py29 grad_all_with_sens = C.GradOperation(get_all=True, sens_param=True)
30 grad_by_list_with_sens = C.GradOperation(get_by_list=True, sens_param=True)
32 grad_with_sens = C.GradOperation(sens_param=True)
203 self.grad = C.GradOperation(get_by_list=True, sens_param=True)
293 grad_op = C.GradOperation(get_all=False, get_by_list=True, sens_param=True)
314 grad_op = C.GradOperation(get_all=True, get_by_list=False, sens_param=True)
/third_party/mindspore/tests/st/pynative/parser/
Dtest_parser_construct.py66 grad = GradOperation(sens_param=True)
87 grad = GradOperation(get_all=True, get_by_list=False, sens_param=False)
/third_party/mindspore/tests/mindspore_test_framework/components/function/
Dcompile_gradient_wrt_inputs.py49 grad_op = GradOperation(get_all=True, sens_param=True)
Drun_gradient_wrt_inputs.py25 grad_op = GradOperation(get_all=True, sens_param=True)
Drun_gradient_wrt_params.py25 grad_op = GradOperation(get_by_list=True, sens_param=True)

1234567