Searched defs:batch_outs (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training_utils_v1.py | 90 def create(self, batch_outs): argument
99 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
134 def create(self, batch_outs): argument
137 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
430 def create(self, batch_outs): argument
455 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distributed_training_utils_v1.py | 1030 def _per_replica_aggregate_batch(strategy, batch_outs, model, mode): argument
|