Searched defs:grad_res (Results 1 – 3 of 3) sorted by relevance
| /external/executorch/extension/training/test/ |
| D | training_loop_test.cpp | 68 auto grad_res = mod.named_gradients("forward"); in TEST_F() local
|
| /external/executorch/extension/training/module/test/ |
| D | training_module_test.cpp | 56 auto grad_res = mod.named_gradients("forward"); in TEST_F() local
|
| /external/pytorch/test/cpp/api/ |
| D | autograd.cpp | 77 auto grad_res = grad({res}, {x, y}, {torch::ones({2, 2})}); in TEST() local
135 auto grad_res = grad({x * 2}, {x, y}, {}, {}, false, true); in TEST() local
173 auto grad_res = torch::autograd::grad({x2}, {y}, {}, {}, false, true); in TEST() local
|