Searched defs:PostAccumulateGradHook (Results 1 – 1 of 1) sorted by relevance
44 struct TORCH_API PostAccumulateGradHook { struct49 virtual void compiled_args(torch::dynamo::autograd::CompiledNodeArgs& args) { in compiled_args()55 virtual void apply_with_saved( in apply_with_saved()