Searched defs:FunctionPostHook (Results 1 – 1 of 1) sorted by relevance
31 struct TORCH_API FunctionPostHook { struct37 virtual void compiled_args(torch::dynamo::autograd::CompiledNodeArgs& args) { in compiled_args()