torch.fx.experimental.graph_gradual_typechecker.conv2d_inference_rule

torch.fx.experimental.graph_gradual_typechecker.conv2d_inference_rule(n, module_instance)

Given a Conv2D instance and a node check the following conditions: - the input type can be expanded to a size 4 tensor: t = (x_1, x_2, H, W) - the current node type can be expanded to a size 4 tensor: t’ = (x_1’, x_2’, x_3’, x_4’) - x_2 is consistent with the module’s in_channels - let o = (x_1, out_channels, H_out, W_out) then the output is the greatest upper bound of o and the existing node type t’.

Return type:

Any