torch.fx.experimental.graph_gradual_typechecker.calculate_out_dimension

torch.fx.experimental.graph_gradual_typechecker.calculate_out_dimension(d_in, module_instance, index)

For calculating h_in and w_out according to the conv2D documentation