Searched refs:materializeGradWrappers (Results 1 – 1 of 1) sorted by relevance
31 static Tensor materializeGradWrappers(const Tensor& tensor, int64_t current_level) { in materializeGradWrappers() function54 return materializeGradWrappers(tensor, level()); in lift()58 return materializeGradWrappers(tensor, level()); in lift()73 return materializeGradWrappers(tensor, current_level); in autogradBasedTransformProcess()