Searched refs:call_collection (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | save_impl.py | 161 call_collection = LayerCallCollection(layer)
162 call_fn_with_losses = call_collection.add_function(
165 call_fn = call_collection.add_function(
175 call_collection.add_function(
508 def layer_call_wrapper(call_collection, method): argument
512 layer = call_collection.layer
514 inputs = call_collection.get_input_arg_value(args, kwargs)
516 if (args or kwargs) and call_collection.training_arg_was_passed(
518 training = call_collection.get_training_arg_value(args, kwargs)
535 def __init__(self, call_collection, python_function, *args, **kwargs): argument
[all …]
|
| D | saved_model_test.py | 937 call_collection = keras_save.LayerCallCollection(layer)
938 fn = call_collection.add_function(layer.call, 'call')
939 fn2 = call_collection.add_function(layer.call2, 'call2')
957 call_collection = keras_save.LayerCallCollection(layer)
958 fn = call_collection.add_function(layer.call, 'call')
1003 call_collection = keras_save.LayerCallCollection(layer)
1004 fn = call_collection.add_function(layer.call, 'call')
|