Searched refs:batch_outs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays_v1.py | 294 batch_outs = f(actual_inputs)
325 if not isinstance(batch_outs, list):
326 batch_outs = [batch_outs]
329 batch_outs = (
331 model._distribution_strategy, batch_outs, model, mode))
335 aggregator.create(batch_outs)
336 aggregator.aggregate(batch_outs)
339 batch_logs = cbks.make_logs(model, batch_logs, batch_outs, mode)
384 batch_outs = f(ins_batch)
385 if not isinstance(batch_outs, list):
[all …]
|
| D | training_utils_v1.py | 96 def create(self, batch_outs): argument
105 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
140 def create(self, batch_outs): argument
141 self.results = [0.] * len(batch_outs)
143 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
146 self.results[0] += batch_outs[0]
148 self.results[0] += batch_outs[0] * (batch_end - batch_start)
150 self.results[1:] = batch_outs[1:]
436 def create(self, batch_outs): argument
440 lambda x: not is_composite_or_composite_value(x), batch_outs)
[all …]
|
| D | training_generator_v1.py | 259 batch_outs = batch_function(*batch_data)
260 if not isinstance(batch_outs, list):
261 batch_outs = [batch_outs]
264 aggregator.create(batch_outs)
281 aggregator.aggregate(batch_outs)
284 batch_logs = cbks.make_logs(model, batch_logs, batch_outs, mode)
|
| D | training_distributed_v1.py | 386 _, batch_outs = K.batch_get_value([test_op, output_tensors])
399 outs[i] += batch_outs[label]
402 outs[i] = batch_outs[label]
533 _, batch_outs = K.batch_get_value([predict_ops, output_tensors])
549 single_model_output = batch_outs[output_start_index:output_end_index]
552 batch_logs = cbks.make_logs(model, batch_logs, batch_outs, mode)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 1040 def _per_replica_aggregate_batch(strategy, batch_outs, model, mode): argument
1046 nested_outs = batch_outs[i * num_replicas:i * num_replicas + num_replicas]
1050 return batch_outs
|