65449110d6d6bec42350c19e2ff907899683783a,scripts/dump-model-params.py,,_import_external_ops,#Any#,16

Before Change

        return
    print("Your graph contains op "{}" which is not loaded into your Tensorflow runtime.".format(op_name))
    print("Therefore the graph cannot be loaded unless you import the relevant libraries first.")
    sys.exit(1)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(

After Change

def _import_external_ops(message):
    if "horovod" in message.lower():
        logger.info("Importing horovod ...")
        import horovod.tensorflow  // noqa
        return
    if  "MaxBytesInUse" in message:
        logger.info("Importing memory_stats ...")
        from tensorflow.contrib.memory_stats import MaxBytesInUse  // noqa
        return
    if "Nccl" in message:
        logger.info("Importing nccl ...")
        if TF_version <= (1, 12):
            try:
                from tensorflow.contrib.nccl.python.ops.nccl_ops import _validate_and_load_nccl_so
            except Exception:

Instances