Fixes for shared queue read functions:

 . fix typo in call
 . re-export the functions so that they be accessed from outside code
 . do not create a new graph for the dummy placement variable. The new
graph doesn't have the device setter applied, so the variable was not
actually placed on the PS.
Change: 137745823

Author: tensorflower-gardener

tensorflow/contrib/learn/python/learn/learn_io/__init__.py

@@ -22,6 +22,8 @@
 from tensorflow.contrib.learn.python.learn.learn_io.dask_io import extract_dask_data
 from tensorflow.contrib.learn.python.learn.learn_io.dask_io import extract_dask_labels
 from tensorflow.contrib.learn.python.learn.learn_io.dask_io import HAS_DASK
+from tensorflow.contrib.learn.python.learn.learn_io.graph_io import _read_keyed_batch_examples_shared_queue
+from tensorflow.contrib.learn.python.learn.learn_io.graph_io import _read_keyed_batch_features_shared_queue
 from tensorflow.contrib.learn.python.learn.learn_io.graph_io import queue_parsed_features
 from tensorflow.contrib.learn.python.learn.learn_io.graph_io import read_batch_examples
 from tensorflow.contrib.learn.python.learn.learn_io.graph_io import read_batch_features

tensorflow/contrib/learn/python/learn/learn_io/graph_io.py

@@ -239,8 +239,7 @@ def _get_shared_file_name_queue(file_names, shuffle, num_epochs, name):
   # Creating a dummy variable so we can put the shared queue in ps if there is
   # a PS and in the worker otherwise. TODO(rohanj): Figure out how to place an
   # op on PS without this hack
-  with ops.Graph().as_default():
-    dummy_var = var_ops.Variable(initial_value=0, name='dummy_var')
+  dummy_var = var_ops.Variable(initial_value=0, name='queue_placement_var')
   with ops.device(dummy_var.device):
     shared_file_name_queue = input_ops.string_input_producer(
         constant_op.constant(
@@ -561,7 +560,7 @@ def _read_keyed_batch_features_shared_queue(file_pattern,
   """
 
   with ops.name_scope(name, 'read_batch_features', [file_pattern]) as scope:
-    keys, examples = read_keyed_batch_examples_shared_queue(
+    keys, examples = _read_keyed_batch_examples_shared_queue(
         file_pattern,
         batch_size,
         reader,