Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:buffer_size

30     buffer_size = array_ops.placeholder(dtypes.int64, shape=[])
32 buffer_size=buffer_size).make_initializable_iterator()
37 sess.run(init_op, feed_dict={buffer_size: 5})
44 buffer_size = array_ops.placeholder(dtypes.int64, shape=[])
46 buffer_size=buffer_size).make_initializable_iterator()
49 with self.assertRaisesRegexp(errors.InvalidArgumentError, "buffer_size"):
51 sess.run(init_op, feed_dict={buffer_size: 0})
53 with self.assertRaisesRegexp(errors.InvalidArgumentError, "buffer_size"):
55 sess.run(init_op, feed_dict={buffer_size: -5})