Searched refs:autograd_kernel_redispatching_with_DispatchKeySet (Results 1 – 1 of 1) sorted by relevance
2048 void autograd_kernel_redispatching_with_DispatchKeySet(c10::DispatchKeySet ks, Tensor a) { in autograd_kernel_redispatching_with_DispatchKeySet() function2075 m.impl("fn", c10::DispatchKey::AutogradCPU, autograd_kernel_redispatching_with_DispatchKeySet); in TEST()2098 m.impl("fn", c10::DispatchKey::AutogradCPU, autograd_kernel_redispatching_with_DispatchKeySet); in TEST()