Searched defs:grad_test_func (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nestedtensor.py | 2921 def grad_test_func(inpt): function
2934 def grad_test_func(tensor, nested_size): function
2950 def grad_test_func(tensor, nested_size): function
2965 def grad_test_func(a, b, c): function
2998 def grad_test_func(a, b, c, d): function
3036 def grad_test_func(a, b, c, d): function
3072 def grad_test_func(a, b): function
3100 def grad_test_func(a, b): function
3146 def grad_test_func(a, b): function
3173 def grad_test_func(a, b): function
[all …]
|