Searched refs:maybeTransformGradWrappers (Results 1 – 1 of 1) sorted by relevance
72 auto maybeTransformGradWrappers = [&](const Tensor& tensor) { in autogradBasedTransformProcess() local76 …t64_t>(stack->size() - num_args), static_cast<int64_t>(stack->size()), maybeTransformGradWrappers); in autogradBasedTransformProcess()