Searched refs:serialized_fns (Results 1 – 1 of 1) sorted by relevance
236 def replace_layer_functions(child_layer, serialized_fns): argument244 child_layer._activity_regularizer = serialized_fns.get(251 serialized_fns['call_and_return_conditional_losses'],254 def replace_metric_functions(child_layer, serialized_fns): argument262 child_layer.__call__ = serialized_fns['__call__']263 child_layer.result = serialized_fns['result']264 child_layer.update_state = serialized_fns['update_state']