67fb07ba9f
A new class LoadOptions is created similar to the existing SavedOptions. The option experimental_io_device is the only option added at this time and usd to set the io_device when loading a SavedModel for distributed training. PiperOrigin-RevId: 316557681 Change-Id: If3f1eae18b09085ff11dc8a6882fabcb18f5f48e
36 lines
1.3 KiB
Plaintext
36 lines
1.3 KiB
Plaintext
path: "tensorflow.keras.models"
|
|
tf_module {
|
|
member {
|
|
name: "Model"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member {
|
|
name: "Sequential"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member_method {
|
|
name: "clone_model"
|
|
argspec: "args=[\'model\', \'input_tensors\', \'clone_function\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "load_model"
|
|
argspec: "args=[\'filepath\', \'custom_objects\', \'compile\', \'options\'], varargs=None, keywords=None, defaults=[\'None\', \'True\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "model_from_config"
|
|
argspec: "args=[\'config\', \'custom_objects\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "model_from_json"
|
|
argspec: "args=[\'json_string\', \'custom_objects\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "model_from_yaml"
|
|
argspec: "args=[\'yaml_string\', \'custom_objects\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "save_model"
|
|
argspec: "args=[\'model\', \'filepath\', \'overwrite\', \'include_optimizer\', \'save_format\', \'signatures\', \'options\'], varargs=None, keywords=None, defaults=[\'True\', \'True\', \'None\', \'None\', \'None\'], "
|
|
}
|
|
}
|