class_docstr: Wrapper of OutOfGraphReplayBuffer with an in graph sampling mechanism.
  
  Usage:
    To add a transition:  call the add function.
  
    To sample a batch:    Construct operations that depend on any of the
                          tensors is the transition dictionary. Every sess.run
                          that requires any of these tensors will sample a new
                          transition.
docstr: Sets up staging ops for prefetching the next transition.
  
  This allows us to hide the py_func latency. To do so we use a staging area
  to pre-fetch the next batch of transitions.
  
  Args:
    transition: tuple of tf.Tensors with shape
      memory.get_transition_elements().
  
  Returns:
    prefetched_transition: tuple of tf.Tensors with shape
      memory.get_transition_elements() that have been previously prefetched.
docstr_tok: ["Sets","up","staging","ops","for","prefetching","the","next","transition",".","This","allows","us","to","hide","the","py_func","latency",".","To","do","so","we","use","a","staging","area","to","pre-fetch","the","next","batch","of","transitions",".","Args",":","transition",":","tuple","of","tf.Tensors","with","shape","memory.get_transition_elements","(",")",".","Returns",":","prefetched_transition",":","tuple","of","tf.Tensors","with","shape","memory.get_transition_elements","(",")","that","have","been","previously","prefetched","."]
code: def _set_up_staging(self, transition):
      """Sets up staging ops for prefetching the next transition.
  
      This allows us to hide the py_func latency. To do so we use a staging area
      to pre-fetch the next batch of transitions.
  
      Args:
        transition: tuple of tf.Tensors with shape
          memory.get_transition_elements().
  
      Returns:
        prefetched_transition: tuple of tf.Tensors with shape
          memory.get_transition_elements() that have been previously prefetched.
      """
      del transition  # Unused.
      raise NotImplementedError
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  import gzip
  import math
  import os
  import pickle
  from absl import logging
  import gin.tf
  import numpy
  import tensorflow
label_desc: Use image node_id 5 for calling the WrappedReplayBuffer obj's underlying member method code with example usage: obj._set_up_staging(transition) without return types