Searched defs:grad_op (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | block_util.py | 245 def __init__(self, network, grad_op, sens): argument
259 def __init__(self, network, grad_op): argument
272 def __init__(self, network, grad_op, sens): argument
287 def __init__(self, network, grad_op): argument
300 def gen_grad_net(net, grad_op, input_num, sens=None, training=True, desc_const=(), argument
328 def create_funcs(verification_set, block_generator, block_runner, grad_op=None, default_rand_func=N… argument
|
| D | bprop_util.py | 30 def __init__(self, func, wrt_params, params, grad_op, sens): argument
|
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_sparse_tensor.py | 36 grad_op = C.GradOperation(get_all=True) variable
|
| /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
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | anf_ir_utils.cc | 217 prim::GradOperationPtr grad_op = meta_func_graph->cast<prim::GradOperationPtr>(); in GetMetaFuncGraphText() local
|