torch.fx.experimental.graph_gradual_typechecker.linear_check#

torch.fx.experimental.graph_gradual_typechecker.linear_check(tensor_type, module_instance)[source]#

Checks that an input tensor type satisfies the conditions for linear operation and returns the output type based on in and out features given by module_instance