Beispiel #1
0
 def fn():
     h = gen_data_flow_ops.stack_v2(5,
                                    dtypes.float32,
                                    stack_name="foo")
     c = gen_data_flow_ops.stack_push_v2(h, v)
     with ops.control_dependencies([c]):
         gen_data_flow_ops.stack_close_v2(h)
Beispiel #2
0
 def _testCloseStack(self, use_gpu):
     with self.cached_session(use_gpu=use_gpu) as sess:
         h = gen_data_flow_ops.stack_v2(-1,
                                        elem_type=dtypes.float32,
                                        stack_name="foo")
         c1 = gen_data_flow_ops.stack_close_v2(h)
         sess.run(c1)
 def _testPushCloseStack(self, use_gpu):
   with self.test_session(use_gpu=use_gpu) as sess:
     h = gen_data_flow_ops.stack_v2(
         -1, elem_type=dtypes.float32, stack_name="foo")
     c = gen_data_flow_ops.stack_push_v2(h, [[4.0, 5.0]])
     with ops.control_dependencies([c]):
       c1 = gen_data_flow_ops.stack_close_v2(h)
     sess.run(c1)
 def testCloseStack(self):
     with self.test_session() as sess, self.test_scope():
         size = array_ops.placeholder(dtypes.int32)
         h = gen_data_flow_ops.stack_v2(size,
                                        dtypes.float32,
                                        stack_name="foo")
         c1 = gen_data_flow_ops.stack_close_v2(h)
         sess.run(c1, {size: 5})
 def testPushCloseStack(self):
     with self.test_session() as sess, self.test_scope():
         v = array_ops.placeholder(dtypes.float32)
         h = gen_data_flow_ops.stack_v2(5, dtypes.float32, stack_name="foo")
         c = gen_data_flow_ops.stack_push_v2(h, v)
         with ops.control_dependencies([c]):
             c1 = gen_data_flow_ops.stack_close_v2(h)
         sess.run(c1, {v: [[4.0, 5.0]]})
 def testPushCloseStack(self):
   with self.test_session() as sess, self.test_scope():
     v = array_ops.placeholder(dtypes.float32)
     h = gen_data_flow_ops.stack_v2(5, dtypes.float32, stack_name="foo")
     c = gen_data_flow_ops.stack_push_v2(h, v)
     with ops.control_dependencies([c]):
       c1 = gen_data_flow_ops.stack_close_v2(h)
     sess.run(c1, {v: [[4.0, 5.0]]})
 def _testCloseStack(self, use_gpu):
   with self.test_session(use_gpu=use_gpu) as sess:
     h = gen_data_flow_ops.stack_v2(
         -1, elem_type=dtypes.float32, stack_name="foo")
     c1 = gen_data_flow_ops.stack_close_v2(h)
     sess.run(c1)
 def testCloseStack(self):
   with self.test_session() as sess, self.test_scope():
     size = array_ops.placeholder(dtypes.int32)
     h = gen_data_flow_ops.stack_v2(size, dtypes.float32, stack_name="foo")
     c1 = gen_data_flow_ops.stack_close_v2(h)
     sess.run(c1, {size: 5})
Beispiel #9
0
 def fn():
     h = gen_data_flow_ops.stack_v2(5,
                                    dtypes.float32,
                                    stack_name="foo")
     gen_data_flow_ops.stack_close_v2(h)