Home
last modified time | relevance | path

Searched defs:grad_spec (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_autodiff.cpp167 auto grad_spec = differentiate(graph); in TEST() local
216 auto grad_spec = differentiate(graph); in TEST() local
263 auto grad_spec = differentiate(g); in TEST() local
H A Dtest_utils.cpp39 Gradient& grad_spec, in runGradient()
H A Dtest_misc.cpp205 auto grad_spec = differentiate(graph); in TEST() local
323 auto grad_spec = differentiate(graph); in TEST() local