Searched refs:ComputeRequiresGrad (Results 1 – 1 of 1) sorted by relevance
39 struct ComputeRequiresGrad : IterArgs<ComputeRequiresGrad> { struct41 using IterArgs<ComputeRequiresGrad>::operator(); argument63 return ComputeRequiresGrad().apply(std::forward<Args>(args)...).out; in compute_requires_grad() argument