diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
| -rw-r--r-- | tensorflow/core/kernels/BUILD | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index ab5923b0e7..06a11d31ab 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -3362,6 +3362,33 @@ tf_cuda_cc_test( ) tf_kernel_library( + name = "random_poisson_op", + prefix = "random_poisson_op", + deps = [ + ":random_ops", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", + "//tensorflow/core:random_ops_op_lib", + ], +) + +tf_cuda_cc_test( + name = "random_poisson_op_test", + size = "small", + srcs = ["random_poisson_op_test.cc"], + deps = [ + ":ops_util", + ":random_poisson_op", + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + ], +) + +tf_kernel_library( name = "word2vec_kernels", prefix = "word2vec_kernels", deps = [ |