d73e36a44ab3fa760d385c881db0fa393a557a65,torch/quantization/ns/numeric_suite_core_apis_fx.py,,prepare_model_with_stubs,#Any#Any#Any#Any#Any#,271

Before Change

    matched_subgraph_pairs = get_matching_subgraph_pairs(gm_a, gm_b)
    gm_a_shadows_b = create_a_shadows_b(
        name_a, gm_a, name_b, gm_b, matched_subgraph_pairs, logger_cls)
    return gm_a_shadows_b

// Note: this is not a user facing API
// TODO(future PR): wrap this in a user facing API which does not
//   expose FX types.

After Change

    Same thing as prepare_model_outputs, but for an `a_shadows_b` model.
    TODO(future PR): real docblock
    
    tracer_a, tracer_b = NSTracer(), NSTracer()
    gm_a = GraphModule(model_a, tracer_a.trace(model_a))
    gm_b = GraphModule(model_b, tracer_b.trace(model_b))
    return _prepare_model_with_stubs_impl(name_a, gm_a, name_b, gm_b, logger_cls)


// TODO(future PR): align on naming
// this is equivalent of just the comparison extraction part of `ns.compare_model_stub`

Instances