Rate this Page

torch.fx.experimental.graph_gradual_typechecker.register_refinement_rule#

torch.fx.experimental.graph_gradual_typechecker.register_refinement_rule(call_target)[source]#
Return type:

Callable[[Callable[[~_P], _T]], Callable[[~_P], _T]]