diff options
author | 2018-02-09 10:45:56 -0800 | |
---|---|---|
committer | 2018-02-09 10:50:26 -0800 | |
commit | 4a54f16a66d8d9b4df74ec810f130edf2373c444 (patch) | |
tree | 36ed8450501d334a2e7da509d01c55f7d361112b /tensorflow/contrib/learn/BUILD | |
parent | 81b698d0dad88b0bb33af720247a3eba1d248e9d (diff) |
regression_head accepts link_fn as parameter, so the use can use self defined link fn
PiperOrigin-RevId: 185161896
Diffstat (limited to 'tensorflow/contrib/learn/BUILD')
-rw-r--r-- | tensorflow/contrib/learn/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/learn/BUILD b/tensorflow/contrib/learn/BUILD index 3c782b54a8..abf6e393bb 100644 --- a/tensorflow/contrib/learn/BUILD +++ b/tensorflow/contrib/learn/BUILD @@ -388,6 +388,7 @@ py_test( "//tensorflow/python:client_testlib", "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:lookup_ops", + "//tensorflow/python:math_ops", "//tensorflow/python:session", "//tensorflow/python:sparse_tensor", "//tensorflow/python:variables", |