Searched refs:call_fn_with_losses (Results 1 – 1 of 1) sorted by relevance
162 call_fn_with_losses = call_collection.add_function(166 _extract_outputs_from_fn(layer, call_fn_with_losses),169 fns = {'call_and_return_conditional_losses': call_fn_with_losses,177 call_fn_with_losses,183 fns['call_and_return_all_conditional_losses'] = call_fn_with_losses591 layer, call_fn_with_losses, activity_regularizer_fn): argument594 outputs, losses = call_fn_with_losses(inputs, *args, **kwargs)