diff options
| author |  Alexandre Passos <apassos@google.com> | 2018-01-23 11:20:29 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-23 11:24:07 -0800 |
| commit | 2a3559feb6564e4e46a56a71b200f6a17afe69e7 (patch) | |
| tree | 0ae6820ca8e962d0de7861281b8d2a6165aeb730 /tensorflow/contrib/batching/python | |
| parent | d1020bfdedadc3da7c89a69651c45cc790922b69 (diff) | |
Moves batch ops to core and exposes tf.contrib.batching
PiperOrigin-RevId: 182963906
Diffstat (limited to 'tensorflow/contrib/batching/python')
| -rw-r--r-- | tensorflow/contrib/batching/python/ops/batch_ops.py | 12 |
1 files changed, 3 insertions, 9 deletions
diff --git a/tensorflow/contrib/batching/python/ops/batch_ops.py b/tensorflow/contrib/batching/python/ops/batch_ops.py index cee4d7b4a9..4e0b3f9af9 100644 --- a/tensorflow/contrib/batching/python/ops/batch_ops.py +++ b/tensorflow/contrib/batching/python/ops/batch_ops.py @@ -18,18 +18,12 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -from tensorflow.contrib.batching.ops import gen_batch_ops +from tensorflow.python.framework import ops +from tensorflow.python.ops import gen_batch_ops # go/tf-wildcard-import # pylint: disable=wildcard-import -from tensorflow.contrib.batching.ops.gen_batch_ops import * +from tensorflow.python.ops.gen_batch_ops import * # pylint: enable=wildcard-import -from tensorflow.contrib.util import loader -from tensorflow.python.framework import ops -from tensorflow.python.platform import resource_loader - - -_batch_ops = loader.load_op_library( - resource_loader.get_path_to_datafile("_batch_ops.so")) @ops.RegisterGradient("Batch") |