diff options
author | 2018-04-23 19:35:19 -0700 | |
---|---|---|
committer | 2018-04-23 19:42:05 -0700 | |
commit | 3f7c9265b59cae306d029dfac76e25badd20def8 (patch) | |
tree | 3128bf6b44fd671c3079934fce3ed748cb0bbaf8 /tensorflow/contrib/coder | |
parent | 7bee86727b87a8317d4f1407061edfa9ccb16ea5 (diff) |
Add missing pmf_to_cdf_op.cc in the source list in cmake.
Also split range_coder_ops.cc and range_coder_ops_util.cc into separate targets
so that dependence to range_coder_ops_util.cc does not register kernels again.
PiperOrigin-RevId: 194024410
Diffstat (limited to 'tensorflow/contrib/coder')
-rw-r--r-- | tensorflow/contrib/coder/BUILD | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/tensorflow/contrib/coder/BUILD b/tensorflow/contrib/coder/BUILD index a146460a9c..a2c6e41303 100644 --- a/tensorflow/contrib/coder/BUILD +++ b/tensorflow/contrib/coder/BUILD @@ -54,19 +54,27 @@ tf_gen_op_libs( ], ) +cc_library( + name = "range_coder_ops_util", + srcs = ["kernels/range_coder_ops_util.cc"], + hdrs = ["kernels/range_coder_ops_util.h"], + visibility = ["//visibility:public"], + deps = [ + "//tensorflow/core:framework", + "//tensorflow/core:lib", + ], +) + tf_kernel_library( name = "range_coder_ops", srcs = [ "kernels/range_coder_ops.cc", - "kernels/range_coder_ops_util.cc", - ], - hdrs = [ - "kernels/range_coder_ops_util.h", ], visibility = ["//visibility:public"], deps = [ ":coder_ops_op_lib", ":range_coder", + ":range_coder_ops_util", "//tensorflow/core:framework", "//tensorflow/core:lib", ], |