Skip to content

Commit

Permalink
Add BUILD file for tensorflow_io.video
Browse files Browse the repository at this point in the history
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
  • Loading branch information
yongtang committed Dec 17, 2018
1 parent 8537264 commit 8056c4d
Showing 1 changed file with 62 additions and 0 deletions.
62 changes: 62 additions & 0 deletions tensorflow_io/video/BUILD
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
licenses(["notice"]) # Apache 2.0

package(default_visibility = ["//visibility:public"])

filegroup(
name = "test_data",
srcs = glob(["python/kernel_tests/testdata/*"]),
)

cc_binary(
name = 'python/ops/_video_ops.so',
srcs = [
"kernels/video_dataset_ops.cc",
"ops/dataset_ops.cc",
],
linkshared = 1,
deps = [
"@local_config_tf//:libtensorflow_framework",
"@local_config_tf//:tf_header_lib",
"@ffmpeg//:ffmpeg",
],
copts = ["-pthread", "-std=c++11", "-D_GLIBCXX_USE_CXX11_ABI=0", "-DNDEBUG"]
)

py_library(
name = "video_ops_py",
srcs = ([
"python/ops/video_dataset_ops.py",
]),
data = [
":python/ops/_video_ops.so"
],
srcs_version = "PY2AND3",
)

py_test(
name = "video_py_test",
srcs = [
"python/kernel_tests/video_test.py"
],
main = "python/kernel_tests/video_test.py",
data = [
":test_data",
],
deps = [
":video_ops_py",
],
srcs_version = "PY2AND3",
)

py_library(
name = "video_py",
srcs = ([
"__init__.py",
"python/__init__.py",
"python/ops/__init__.py",
]),
deps = [
":video_ops_py"
],
srcs_version = "PY2AND3",
)

0 comments on commit 8056c4d

Please sign in to comment.