From 0a2a5acbe38df2d1f869aa8d46167ad6d67f3da6 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 3 Jan 2018 08:18:13 -0800 Subject: K-FAC: Add "OnehotCategoricalLogitsNegativeLogProbLoss" to kfac.loss_functions_lib._allowed_symbols. PiperOrigin-RevId: 180672608 --- tensorflow/contrib/kfac/python/ops/loss_functions_lib.py | 1 + 1 file changed, 1 insertion(+) (limited to 'tensorflow/contrib/kfac') 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", -- cgit v1.2.3