diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/boosted_trees/BUILD')
-rw-r--r-- | tensorflow/python/kernel_tests/boosted_trees/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/boosted_trees/BUILD b/tensorflow/python/kernel_tests/boosted_trees/BUILD index 4f92ab0795..20446781f0 100644 --- a/tensorflow/python/kernel_tests/boosted_trees/BUILD +++ b/tensorflow/python/kernel_tests/boosted_trees/BUILD @@ -74,3 +74,16 @@ tf_py_test( "//tensorflow/python:resources", ], ) + +tf_py_test( + name = "quantile_ops_test", + size = "small", + srcs = ["quantile_ops_test.py"], + additional_deps = [ + "//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py", + "//tensorflow/python:boosted_trees_ops", + "//tensorflow/python:constant_op", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:resources", + ], +) |