259ffa9ea6
The current tracing implementation has a limited scope of models and layers that can be traced. When users add a custom layer or model that is unsupported (e.g. with multiple tensor arguments), they'll come across an error that prevents them from saving the model entirely. With this option, those models can now be saved to the SavedModel format for serving or retraining. PiperOrigin-RevId: 338295627 Change-Id: Ieea88ecaa1b8665df4ab45c96e882867e4308d88
357 lines
12 KiB
Plaintext
357 lines
12 KiB
Plaintext
path: "tensorflow.keras.experimental.WideDeepModel"
|
|
tf_class {
|
|
is_instance: "<class \'tensorflow.python.keras.premade.wide_deep.WideDeepModel\'>"
|
|
is_instance: "<class \'tensorflow.python.keras.engine.training.Model\'>"
|
|
is_instance: "<class \'tensorflow.python.keras.engine.base_layer.Layer\'>"
|
|
is_instance: "<class \'tensorflow.python.module.module.Module\'>"
|
|
is_instance: "<class \'tensorflow.python.training.tracking.tracking.AutoTrackable\'>"
|
|
is_instance: "<class \'tensorflow.python.training.tracking.base.Trackable\'>"
|
|
is_instance: "<class \'tensorflow.python.keras.utils.version_utils.LayerVersionSelector\'>"
|
|
is_instance: "<class \'tensorflow.python.keras.utils.version_utils.ModelVersionSelector\'>"
|
|
is_instance: "<type \'object\'>"
|
|
member {
|
|
name: "activity_regularizer"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "compute_dtype"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "distribute_strategy"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "dtype"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "dtype_policy"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "dynamic"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "inbound_nodes"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "input"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "input_mask"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "input_shape"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "input_spec"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "layers"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "losses"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "metrics"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "metrics_names"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "name"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "name_scope"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "non_trainable_variables"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "non_trainable_weights"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "outbound_nodes"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "output"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "output_mask"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "output_shape"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "run_eagerly"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "state_updates"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "stateful"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "submodules"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "supports_masking"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "trainable"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "trainable_variables"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "trainable_weights"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "updates"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "variable_dtype"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "variables"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "weights"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member_method {
|
|
name: "__init__"
|
|
argspec: "args=[\'self\', \'linear_model\', \'dnn_model\', \'activation\'], varargs=None, keywords=kwargs, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "add_loss"
|
|
argspec: "args=[\'self\', \'losses\'], varargs=None, keywords=kwargs, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "add_metric"
|
|
argspec: "args=[\'self\', \'value\', \'name\'], varargs=None, keywords=kwargs, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "add_update"
|
|
argspec: "args=[\'self\', \'updates\', \'inputs\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "add_variable"
|
|
argspec: "args=[\'self\'], varargs=args, keywords=kwargs, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "add_weight"
|
|
argspec: "args=[\'self\', \'name\', \'shape\', \'dtype\', \'initializer\', \'regularizer\', \'trainable\', \'constraint\', \'use_resource\', \'synchronization\', \'aggregation\'], varargs=None, keywords=kwargs, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\', \'None\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
|
|
}
|
|
member_method {
|
|
name: "apply"
|
|
argspec: "args=[\'self\', \'inputs\'], varargs=args, keywords=kwargs, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "build"
|
|
argspec: "args=[\'self\', \'input_shape\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "call"
|
|
argspec: "args=[\'self\', \'inputs\', \'training\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "compile"
|
|
argspec: "args=[\'self\', \'optimizer\', \'loss\', \'metrics\', \'loss_weights\', \'weighted_metrics\', \'run_eagerly\', \'steps_per_execution\'], varargs=None, keywords=kwargs, defaults=[\'rmsprop\', \'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "compute_mask"
|
|
argspec: "args=[\'self\', \'inputs\', \'mask\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "compute_output_shape"
|
|
argspec: "args=[\'self\', \'input_shape\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "compute_output_signature"
|
|
argspec: "args=[\'self\', \'input_signature\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "count_params"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "evaluate"
|
|
argspec: "args=[\'self\', \'x\', \'y\', \'batch_size\', \'verbose\', \'sample_weight\', \'steps\', \'callbacks\', \'max_queue_size\', \'workers\', \'use_multiprocessing\', \'return_dict\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'1\', \'None\', \'None\', \'None\', \'10\', \'1\', \'False\', \'False\'], "
|
|
}
|
|
member_method {
|
|
name: "evaluate_generator"
|
|
argspec: "args=[\'self\', \'generator\', \'steps\', \'callbacks\', \'max_queue_size\', \'workers\', \'use_multiprocessing\', \'verbose\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'10\', \'1\', \'False\', \'0\'], "
|
|
}
|
|
member_method {
|
|
name: "fit"
|
|
argspec: "args=[\'self\', \'x\', \'y\', \'batch_size\', \'epochs\', \'verbose\', \'callbacks\', \'validation_split\', \'validation_data\', \'shuffle\', \'class_weight\', \'sample_weight\', \'initial_epoch\', \'steps_per_epoch\', \'validation_steps\', \'validation_batch_size\', \'validation_freq\', \'max_queue_size\', \'workers\', \'use_multiprocessing\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'1\', \'1\', \'None\', \'0.0\', \'None\', \'True\', \'None\', \'None\', \'0\', \'None\', \'None\', \'None\', \'1\', \'10\', \'1\', \'False\'], "
|
|
}
|
|
member_method {
|
|
name: "fit_generator"
|
|
argspec: "args=[\'self\', \'generator\', \'steps_per_epoch\', \'epochs\', \'verbose\', \'callbacks\', \'validation_data\', \'validation_steps\', \'validation_freq\', \'class_weight\', \'max_queue_size\', \'workers\', \'use_multiprocessing\', \'shuffle\', \'initial_epoch\'], varargs=None, keywords=None, defaults=[\'None\', \'1\', \'1\', \'None\', \'None\', \'None\', \'1\', \'None\', \'10\', \'1\', \'False\', \'True\', \'0\'], "
|
|
}
|
|
member_method {
|
|
name: "from_config"
|
|
argspec: "args=[\'cls\', \'config\', \'custom_objects\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "get_config"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_input_at"
|
|
argspec: "args=[\'self\', \'node_index\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_input_mask_at"
|
|
argspec: "args=[\'self\', \'node_index\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_input_shape_at"
|
|
argspec: "args=[\'self\', \'node_index\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_layer"
|
|
argspec: "args=[\'self\', \'name\', \'index\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "get_losses_for"
|
|
argspec: "args=[\'self\', \'inputs\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_output_at"
|
|
argspec: "args=[\'self\', \'node_index\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_output_mask_at"
|
|
argspec: "args=[\'self\', \'node_index\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_output_shape_at"
|
|
argspec: "args=[\'self\', \'node_index\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_updates_for"
|
|
argspec: "args=[\'self\', \'inputs\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_weights"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "load_weights"
|
|
argspec: "args=[\'self\', \'filepath\', \'by_name\', \'skip_mismatch\', \'options\'], varargs=None, keywords=None, defaults=[\'False\', \'False\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "make_predict_function"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "make_test_function"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "make_train_function"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "predict"
|
|
argspec: "args=[\'self\', \'x\', \'batch_size\', \'verbose\', \'steps\', \'callbacks\', \'max_queue_size\', \'workers\', \'use_multiprocessing\'], varargs=None, keywords=None, defaults=[\'None\', \'0\', \'None\', \'None\', \'10\', \'1\', \'False\'], "
|
|
}
|
|
member_method {
|
|
name: "predict_generator"
|
|
argspec: "args=[\'self\', \'generator\', \'steps\', \'callbacks\', \'max_queue_size\', \'workers\', \'use_multiprocessing\', \'verbose\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'10\', \'1\', \'False\', \'0\'], "
|
|
}
|
|
member_method {
|
|
name: "predict_on_batch"
|
|
argspec: "args=[\'self\', \'x\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "predict_step"
|
|
argspec: "args=[\'self\', \'data\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "reset_metrics"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "reset_states"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "save"
|
|
argspec: "args=[\'self\', \'filepath\', \'overwrite\', \'include_optimizer\', \'save_format\', \'signatures\', \'options\', \'save_traces\'], varargs=None, keywords=None, defaults=[\'True\', \'True\', \'None\', \'None\', \'None\', \'True\'], "
|
|
}
|
|
member_method {
|
|
name: "save_weights"
|
|
argspec: "args=[\'self\', \'filepath\', \'overwrite\', \'save_format\', \'options\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "set_weights"
|
|
argspec: "args=[\'self\', \'weights\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "summary"
|
|
argspec: "args=[\'self\', \'line_length\', \'positions\', \'print_fn\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
|
|
}
|
|
member_method {
|
|
name: "test_on_batch"
|
|
argspec: "args=[\'self\', \'x\', \'y\', \'sample_weight\', \'reset_metrics\', \'return_dict\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'True\', \'False\'], "
|
|
}
|
|
member_method {
|
|
name: "test_step"
|
|
argspec: "args=[\'self\', \'data\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "to_json"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=kwargs, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "to_yaml"
|
|
argspec: "args=[\'self\'], varargs=None, keywords=kwargs, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "train_on_batch"
|
|
argspec: "args=[\'self\', \'x\', \'y\', \'sample_weight\', \'class_weight\', \'reset_metrics\', \'return_dict\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'True\', \'False\'], "
|
|
}
|
|
member_method {
|
|
name: "train_step"
|
|
argspec: "args=[\'self\', \'data\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "with_name_scope"
|
|
argspec: "args=[\'cls\', \'method\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
}
|