Searched refs:flatten_per_replica_values (Results 1 – 2 of 2) sorted by relevance
113 all_inputs = flatten_per_replica_values(distribution_strategy,119 all_updates = flatten_per_replica_values(distribution_strategy,128 all_session_args['feed_dict'] = flatten_per_replica_values(133 all_session_args['fetches'] = flatten_per_replica_values(143 return flatten_per_replica_values(strategy, grouped_outputs)150 output_losses = flatten_per_replica_values(strategy,152 metrics = flatten_per_replica_values(strategy,194 return flatten_per_replica_values(distribution_strategy,202 all_outputs = flatten_per_replica_values(distribution_strategy,216 def flatten_per_replica_values(distribution_strategy, per_replica_values): function[all …]
494 output_tensors = dist_utils.flatten_per_replica_values(