Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dgrad_var_prepare.cc33 … const AnfNodePtr &func_node, bool is_unpack, bool sens_param) { in GenerateUnpackGraphNode()
134 const bool sens_param = grad_op_ptr->sens_param(); in operator ()() local
/third_party/mindspore/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
Dtest_multi_grad.py60 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
70 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/tests/st/pynative/
Dtest_graph_param_transform.py60 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
70 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py51 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
157 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
Dtest_pynative_hook_grad.py92 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/tests/st/high_grad/
Dtest_highgrad_train.py58 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/tests/st/ops/ascend/
Dtest_rnn_op.py28 def __init__(self, network, sens_param): argument
Dtest_gru_op.py28 def __init__(self, network, sens_param): argument
Dtest_lstm_op.py29 def __init__(self, network, sens_param): argument
/third_party/mindspore/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
616 def __init__(self, network, grad_list, sens_param=False, real_inputs_count=None): argument
Dtest_auto_monad_gpu.py65 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
75 def __init__(self, network, sens_param=True, real_inputs_count=None): argument
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.h155 bool sens_param() const { return sens_param_; } in sens_param() function
Dcomposite.cc562 …dOperation::GradOperation(const std::string &name, bool get_all, bool get_by_list, bool sens_param) in GradOperation()
/third_party/mindspore/mindspore/ops/composite/
Dbase.py319 def __init__(self, get_all=False, get_by_list=False, sens_param=False): argument
/third_party/mindspore/mindspore/nn/wrap/
Dcell_wrapper.py250 def __init__(self, network, weights=None, get_all=False, get_by_list=False, sens_param=False): argument
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkpynative.cc980 auto sens_param = tape_->add_parameter(); in SetSensAndWeights() local