Searched refs:original_model (Results 1 – 2 of 2) sorted by relevance
846 def _copy_weights_to_distributed_model(original_model, mode): argument848 strategy = original_model._distribution_strategy849 distributed_model = get_distributed_model(original_model, mode)853 orig_model_weights = original_model.get_weights()
309 def chained_model_outputs(original_model, data): argument315 state_manager.initialize_graph(original_model)317 model=original_model,