Searched defs:FuncTorchGradWrapper (Results 1 – 3 of 3) sorted by relevance
207 TORCH_LIBRARY_IMPL(_, FuncTorchGradWrapper, m) { in TORCH_LIBRARY_IMPL() argument
386 FuncTorchGradWrapper, // See Note [Out-of-tree vmap+grad prototype] enumerator
117 FuncTorchGradWrapper = auto() variable in DispatchKey