Searched refs:grouped_outputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 83 def unwrap_values(distribution_strategy, grouped_inputs, grouped_outputs, argument
115 all_outputs = unwrap_outputs(distribution_strategy, grouped_outputs,
140 def unwrap_output_dict(strategy, grouped_outputs, mode): argument
143 return flatten_per_replica_values(strategy, grouped_outputs)
149 grouped_outputs['total_loss'][0], axis=None)
151 grouped_outputs['output_losses'])
153 grouped_outputs['metrics'])
155 grouped_outputs['batch_size'], axis=None)
172 def unwrap_outputs(distribution_strategy, grouped_outputs, argument
195 grouped_outputs)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed_v1.py | 90 (grouped_inputs, grouped_outputs, grouped_updates,
96 grouped_outputs,
|