Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:count_placeholder

37     count_placeholder = array_ops.placeholder(dtypes.int64, shape=[])
40 .repeat(count_placeholder).make_initializable_iterator())
49 sess.run(init_op, feed_dict={count_placeholder: 3})
59 sess.run(init_op, feed_dict={count_placeholder: 7})
68 sess.run(init_op, feed_dict={count_placeholder: 0})
75 sess.run(init_op, feed_dict={count_placeholder: -1})
83 count_placeholder = array_ops.placeholder(dtypes.int64, shape=[])
86 .take(count_placeholder).make_initializable_iterator())
95 sess.run(init_op, feed_dict={count_placeholder: 4})
104 sess.run(init_op, feed_dict={count_placeholder: 25})
113 sess.run(init_op, feed_dict={count_placeholder: -1})
122 sess.run(init_op, feed_dict={count_placeholder: 0})
129 count_placeholder = array_ops.placeholder(dtypes.int64, shape=[])
132 .skip(count_placeholder).make_initializable_iterator())
142 sess.run(init_op, feed_dict={count_placeholder: 4})
150 sess.run(init_op, feed_dict={count_placeholder: 25})
155 sess.run(init_op, feed_dict={count_placeholder: 10})
160 sess.run(init_op, feed_dict={count_placeholder: -1})
165 sess.run(init_op, feed_dict={count_placeholder: 0})