tf.parallel_stack