Home
last modified time | relevance | path

Searched refs:materializeGradWrappers (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/functorch/
DADInterpreters.cpp31 static Tensor materializeGradWrappers(const Tensor& tensor, int64_t current_level) { in materializeGradWrappers() function
54 return materializeGradWrappers(tensor, level()); in lift()
58 return materializeGradWrappers(tensor, level()); in lift()
73 return materializeGradWrappers(tensor, current_level); in autogradBasedTransformProcess()