diff options
| author |  Shivani Agrawal <shivaniagrawal@google.com> | 2018-10-02 17:48:25 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-02 17:52:21 -0700 |
| commit | b7e9cbab27c893283acc4a6154d7a59dffb23758 (patch) | |
| tree | 2eed3086deb2390f8bf6b8074ddd654ca59cfec8 /tensorflow/contrib/distribute | |
| parent | 80821abd6410f47130fc031b15e9ac220de5b1b9 (diff) | |
Use `defun` instead of `Defun` for `tf.data`, except for `make_one_shot_iterator` which is to be deprecated in future.
PiperOrigin-RevId: 215491729
Diffstat (limited to 'tensorflow/contrib/distribute')
| -rw-r--r-- | tensorflow/contrib/distribute/python/input_ops.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/distribute/python/input_ops.py b/tensorflow/contrib/distribute/python/input_ops.py index f07ec8234d..423952c9e2 100644 --- a/tensorflow/contrib/distribute/python/input_ops.py +++ b/tensorflow/contrib/distribute/python/input_ops.py @@ -78,7 +78,7 @@ def auto_shard_dataset(dataset, num_shards, index): elif hasattr(dataset, "_map_func"): # TODO(priyag): Make this check more robust by enforcing some common # property on all map/flatmap/interleave datasets. - map_func_def = dataset._map_func.definition + map_func_def = dataset._map_func.function_def for node in map_func_def.node_def: if node.op in _READER_DATASET_OPS: found_reader_op = True |