Searched defs:jac_wrap_with_jit_function (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/gradient/ | ||
| D | test_jacfwd.py | 54 def jac_wrap_with_jit_function(x, y, z): function |
| D | test_jacrev.py | 54 def jac_wrap_with_jit_function(x, y, z): function |