Searched refs:dist_model (Results 1 – 1 of 1) sorted by relevance
46 def set_weights(distribution_strategy, dist_model, weights): argument60 for layer in dist_model.layers: