Searched defs:call_fn (Results 1 – 2 of 2) sorted by relevance
41 def use_wrapped_call(layer, call_fn, default_training_value=None, argument203 def get_training_arg_index(call_fn): argument
456 def _maybe_wrap_with_training_arg(self, call_fn): argument494 def add_function(self, call_fn, name): argument