Home
last modified time | relevance | path

Searched defs:sens_param (Results 1 – 25 of 32) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/tests/st/pynative/utils/
Dgrad.py51 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
60 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
69 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
78 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
89 def __init__(self, network, sens_param=False, real_inputs_count=None): argument
99 def __init__(self, network, grad_list, sens_param=False, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/share/
Dgrad.py62 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
71 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
80 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
89 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
99 def __init__(self, network, grad_list, sens_param=False, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dgrad.py77 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
87 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
Dtest_gru_op.py28 def __init__(self, network, sens_param): argument
Dtest_rnn_op.py28 def __init__(self, network, sens_param): argument
Dtest_lstm_op.py28 def __init__(self, network, sens_param): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_sequence/
Dsequence_help.py48 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
57 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/pynative_mode/
Dtest_graph_param_cases.py62 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
72 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/st/ge/ge_pass/
Dpass_dropout.py48 def __init__(self, network, sens_param=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/pynative/
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py50 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
156 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/
Dtest_gru.py63 def __init__(self, network, sens_param): argument
/third_party/mindspore/mindspore-src/source/tests/st/auto_monad/
Dtest_auto_monad_mindtester.py61 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
71 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
81 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
672 def __init__(self, network, grad_list, sens_param=False, real_inputs_count=None): argument
Dtest_auto_monad_gpu.py62 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/pynative_mode/grad/
Dtest_multi_grad.py61 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
71 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_lstm_op.py28 def __init__(self, network, sens_param): argument
/third_party/mindspore/mindspore-src/source/tests/st/high_grad/
Dtest_highgrad_train.py59 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_lstm_op.py29 def __init__(self, network, sens_param): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/
Dtest_dynamic_setitem.py55 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
65 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore-src/source/tests/st/rl/
Dtest_cell_list_getattr.py44 def __init__(self, network, get_all=False, get_by_list=False, sens_param=False): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.h194 bool sens_param() const { return sens_param_; } in sens_param() function
/third_party/mindspore/mindspore-src/source/tests/st/pynative/grad/
Dtest_common_grad.py235 def __init__(self, net_me, get_all, get_by_list, sens_param, net_params=None, argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/
Dbase.py328 def __init__(self, get_all=False, get_by_list=False, sens_param=False): argument
517 …def __init__(self, get_all=False, get_by_list=False, sens_param=False, get_by_position=False, has_… argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/wrap/
Dcell_wrapper.py281 def __init__(self, network, weights=None, get_all=False, get_by_list=False, sens_param=False): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/
Dspecial_func_infer.cc392 void HandleGradFuncCall(CallNode *call_node, AObject *decorated, bool sens_param, const py::object … in HandleGradFuncCall()
462 bool sens_param = grad.attr("sens_param").ptr() == Py_True; in HandleGradFunc() local
/third_party/mindspore/mindspore-src/source/tests/st/mutable/
Dtest_grad_mutable.py659 def __init__(self, network, get_all=False, get_by_list=False, sens_param=False): argument

12