GraphTypeChecker

class torch.fx.experimental.graph_gradual_typechecker.GraphTypeChecker(env, traced)

type_check()

A gradual type checker for graphs Effect: every node’s field type will be populated with a type after type-checking is done

Return type:

bool

type_check_node(n)

Type check a given fx node. Current operations: - Reshape - Transpose - Add - Relu - conv2d - batchnorm2d - flatten - maxpool2d - adaptiveavgpool2d - linear

Return type:

Any