diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-03 08:18:13 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-03 08:22:07 -0800 |
commit | 0a2a5acbe38df2d1f869aa8d46167ad6d67f3da6 (patch) | |
tree | 93def831edc46f6be27975f11a94de681d0e8f1b /tensorflow/contrib/kfac | |
parent | a0eb84fe521103bdee6278d917a4f27686781795 (diff) |
K-FAC: Add "OnehotCategoricalLogitsNegativeLogProbLoss" to kfac.loss_functions_lib._allowed_symbols.
PiperOrigin-RevId: 180672608
Diffstat (limited to 'tensorflow/contrib/kfac')
-rw-r--r-- | tensorflow/contrib/kfac/python/ops/loss_functions_lib.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/kfac/python/ops/loss_functions_lib.py b/tensorflow/contrib/kfac/python/ops/loss_functions_lib.py index e9bb4f14e9..705a871d48 100644 --- a/tensorflow/contrib/kfac/python/ops/loss_functions_lib.py +++ b/tensorflow/contrib/kfac/python/ops/loss_functions_lib.py @@ -31,6 +31,7 @@ _allowed_symbols = [ "NormalMeanNegativeLogProbLoss", "NormalMeanVarianceNegativeLogProbLoss", "CategoricalLogitsNegativeLogProbLoss", + "OnehotCategoricalLogitsNegativeLogProbLoss", "MultiBernoulliNegativeLogProbLoss", "MultiBernoulliNegativeLogProbLoss", "insert_slice_in_zeros", |