Searched defs:call_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | utils.py | 38 def use_wrapped_call(layer, call_fn, default_training_value=None, argument 199 def get_training_arg_index(call_fn): argument
|
| D | save_impl.py | 508 def _maybe_wrap_with_training_arg(self, call_fn, match_layer_training_arg): argument 547 def add_function(self, call_fn, name, match_layer_training_arg): argument 614 def __init__(self, call_collection, call_fn, name, input_signature): argument
|