Home
last modified time | relevance | path

Searched refs:ReductionV2 (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/keras/
Dlosses.py90 def __init__(self, reduction=losses_utils.ReductionV2.AUTO, name=None):
105 losses_utils.ReductionV2.validate(reduction)
198 (self.reduction == losses_utils.ReductionV2.AUTO or
199 self.reduction == losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE)):
214 if self.reduction == losses_utils.ReductionV2.AUTO:
215 return losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE
224 reduction=losses_utils.ReductionV2.AUTO,
311 reduction=losses_utils.ReductionV2.AUTO,
370 reduction=losses_utils.ReductionV2.AUTO,
430 reduction=losses_utils.ReductionV2.AUTO,
[all …]
Dlosses_test.py195 self.assertEqual(mse_obj.reduction, losses_utils.ReductionV2.AUTO)
310 reduction=losses_utils.ReductionV2.SUM, name='mse_1')
312 self.assertEqual(mse_obj.reduction, losses_utils.ReductionV2.SUM)
392 mse_obj = losses.MeanSquaredError(reduction=losses_utils.ReductionV2.NONE)
402 mse_obj = losses.MeanSquaredError(reduction=losses_utils.ReductionV2.SUM)
416 reduction=losses_utils.ReductionV2.SUM, name='mae_1')
418 self.assertEqual(mae_obj.reduction, losses_utils.ReductionV2.SUM)
484 mae_obj = losses.MeanAbsoluteError(reduction=losses_utils.ReductionV2.NONE)
494 mae_obj = losses.MeanAbsoluteError(reduction=losses_utils.ReductionV2.SUM)
519 reduction=losses_utils.ReductionV2.SUM, name='mape_1')
[all …]
Dmetrics_correctness_test.py560 losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE,
561 losses_utils.ReductionV2.AUTO,
562 losses_utils.ReductionV2.SUM
618 losses_utils.ReductionV2.NONE:
620 losses_utils.ReductionV2.SUM: {
625 losses_utils.ReductionV2.AUTO:
627 losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE:
633 losses_utils.ReductionV2.NONE: [144, 56, 88],
634 losses_utils.ReductionV2.SUM: [198, 82, 116],
635 losses_utils.ReductionV2.AUTO: [144, 56, 88],
[all …]
/external/tensorflow/tensorflow/python/keras/utils/
Dlosses_utils.py33 class ReductionV2(object): class
257 reduction=ReductionV2.SUM_OVER_BATCH_SIZE):
259 if reduction == ReductionV2.NONE:
263 if reduction == ReductionV2.SUM_OVER_BATCH_SIZE:
270 reduction=ReductionV2.SUM_OVER_BATCH_SIZE,
289 ReductionV2.validate(reduction)
293 if reduction == ReductionV2.AUTO:
294 reduction = ReductionV2.SUM_OVER_BATCH_SIZE
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_eager_v1.py183 reduction=losses_utils.ReductionV2.NONE)
188 if loss_reduction == losses_utils.ReductionV2.AUTO:
189 loss_reduction = losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE
201 loss_reduction = losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE
213 if loss_reduction == losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE:
Dcompile_utils.py209 if loss_obj.reduction == losses_utils.ReductionV2.SUM:
225 if (loss_obj.reduction == losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE or
226 loss_obj.reduction == losses_utils.ReductionV2.AUTO):
Dtraining_v1.py1614 reduction=losses_utils.ReductionV2.NONE)
1619 if loss_reduction == losses_utils.ReductionV2.AUTO:
1620 loss_reduction = losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE
1632 loss_reduction = losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE
1640 if loss_reduction == losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE:
Dtraining_utils_v1.py1229 reduction=losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.losses.-reduction.pbtxt3 is_instance: "<class \'tensorflow.python.keras.utils.losses_utils.ReductionV2\'>"
Dtensorflow.keras.losses.-reduction.pbtxt3 is_instance: "<class \'tensorflow.python.keras.utils.losses_utils.ReductionV2\'>"
/external/tensorflow/tensorflow/python/keras/saving/
Dlosses_serialization_test.py48 reduction=losses_utils.ReductionV2.AUTO,
/external/tensorflow/tensorflow/python/keras/distribute/
Dparameter_server_training_test.py184 reduction=losses_utils.ReductionV2.NONE)(labels, pred))
Ddistribute_strategy_test.py2064 losses_utils.ReductionV2.AUTO,
2065 losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE,
2066 losses_utils.ReductionV2.SUM
2569 reduction=losses_utils.ReductionV2.SUM_OVER_BATCH_SIZE),