diff options
author | 2017-03-23 16:36:51 -0800 | |
---|---|---|
committer | 2017-03-23 17:52:19 -0700 | |
commit | b2ac11c0797447eb36d427a0907fb2f05af57fcc (patch) | |
tree | eafeef30eed4451bc55c421d966b01d2fda96327 /tensorflow/contrib/input_pipeline | |
parent | 494fe43926f8928bb3b10c1729abea9be5113f21 (diff) |
Internal changes
Change: 151077399
Diffstat (limited to 'tensorflow/contrib/input_pipeline')
-rw-r--r-- | tensorflow/contrib/input_pipeline/BUILD | 9 | ||||
-rw-r--r-- | tensorflow/contrib/input_pipeline/python/ops/input_pipeline_ops.py | 3 |
2 files changed, 9 insertions, 3 deletions
diff --git a/tensorflow/contrib/input_pipeline/BUILD b/tensorflow/contrib/input_pipeline/BUILD index f3db94c213..9aa5763efc 100644 --- a/tensorflow/contrib/input_pipeline/BUILD +++ b/tensorflow/contrib/input_pipeline/BUILD @@ -21,6 +21,7 @@ load( "//tensorflow/core:platform/default/build_config.bzl", "tf_kernel_tests_linkstatic", ) +load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library") tf_custom_op_library( # TODO(sibyl-Mooth6ku,ptucker): Understand why 'python/ops/_' is needed and fix it. @@ -51,10 +52,14 @@ tf_kernel_library( alwayslink = 1, ) -py_library( +tf_custom_op_py_library( name = "input_pipeline_py", srcs = glob(["python/ops/*.py"]) + ["__init__.py"], - data = [":python/ops/_input_pipeline_ops.so"], + dso = [":python/ops/_input_pipeline_ops.so"], + kernels = [ + ":input_pipeline_ops_kernels", + ":input_pipeline_ops_op_lib", + ], srcs_version = "PY2AND3", deps = [ ":input_pipeline_ops", diff --git a/tensorflow/contrib/input_pipeline/python/ops/input_pipeline_ops.py b/tensorflow/contrib/input_pipeline/python/ops/input_pipeline_ops.py index 101e400e63..be22b5c579 100644 --- a/tensorflow/contrib/input_pipeline/python/ops/input_pipeline_ops.py +++ b/tensorflow/contrib/input_pipeline/python/ops/input_pipeline_ops.py @@ -19,6 +19,7 @@ from __future__ import print_function import random +from tensorflow.contrib.input_pipeline.ops import gen_input_pipeline_ops from tensorflow.contrib.util import loader from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes @@ -42,7 +43,7 @@ def obtain_next(string_list_tensor, counter): An op that produces the element at counter + 1 in the list, round robin style. """ - return _input_pipeline_ops.obtain_next(string_list_tensor, counter) + return gen_input_pipeline_ops.obtain_next(string_list_tensor, counter) def _maybe_randomize_list(string_list, shuffle): |