Searched refs:loss_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.estimator.-nan-tensor-hook.pbtxt | 8 …argspec: "args=[\'self\', \'loss_tensor\', \'fail_on_nan_loss\'], varargs=None, keywords=None, def…
|
| D | tensorflow.train.-nan-tensor-hook.pbtxt | 8 …argspec: "args=[\'self\', \'loss_tensor\', \'fail_on_nan_loss\'], varargs=None, keywords=None, def…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.estimator.-nan-tensor-hook.pbtxt | 8 …argspec: "args=[\'self\', \'loss_tensor\', \'fail_on_nan_loss\'], varargs=None, keywords=None, def…
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | base.py | 297 loss_tensor = regularizer()
298 if loss_tensor is not None:
299 new_losses.append(loss_tensor)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 530 loss_tensor = distribution.reduce(reduce_util.ReduceOp.MEAN, loss_output,
535 loss_tensor = unwrapped_output[0]
536 self.assertEqual(initial_loss.dtype, loss_tensor.dtype)
537 self.assertEqual(initial_loss.shape, loss_tensor.shape)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | basic_session_run_hooks.py | 753 def __init__(self, loss_tensor, fail_on_nan_loss=True): argument
760 self._loss_tensor = loss_tensor
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_v1.py | 955 loss_tensor = regularizer()
956 if loss_tensor is not None:
957 collected_losses.append(loss_tensor)
|
| D | base_layer.py | 1488 loss_tensor = regularizer()
1489 if loss_tensor is not None:
1490 collected_losses.append(loss_tensor)
|