| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | bucket_ops_test.py | 324 allow_small_batch, argument
361 allow_smaller_final_batch=allow_small_batch,
364 expected_batch_size = None if allow_small_batch else batch_size
378 if allow_small_batch:
398 if allow_small_batch:
429 self._testBucketBySequenceLength(allow_small_batch=False)
432 self._testBucketBySequenceLength(allow_small_batch=True)
437 self._testBucketBySequenceLength(allow_small_batch=False,
441 self._testBucketBySequenceLength(allow_small_batch=True,
445 self._testBucketBySequenceLength(allow_small_batch=True,
|
| D | sequence_queueing_state_saver.py | 737 allow_small_batch=False, argument
883 self._allow_small_batch = allow_small_batch
1276 allow_small_batch=True, argument
1525 allow_small_batch=allow_small_batch)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fifo_queue.cc | 196 bool allow_small_batch, in TryDequeueMany() argument
258 [callback, allow_small_batch, in TryDequeueMany()
286 if (allow_small_batch && !queues_[0].empty()) { in TryDequeueMany()
292 if (allow_small_batch) { in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 58 bool allow_small_batch,
275 bool allow_small_batch, in TryDequeueMany() argument
337 [callback, allow_small_batch, in TryDequeueMany()
363 if (allow_small_batch && !queues_[0].empty()) { in TryDequeueMany()
369 if (allow_small_batch) { in TryDequeueMany()
|
| D | priority_queue.cc | 239 bool allow_small_batch, in TryDequeueMany() argument
302 allow_small_batch](Attempt* attempt) EXCLUSIVE_LOCKS_REQUIRED(mu_) { in TryDequeueMany()
310 (!allow_small_batch && s < attempt->elements_requested)) { in TryDequeueMany()
334 if (s == 0 || !(closed_ && allow_small_batch)) { in TryDequeueMany()
|
| D | padding_fifo_queue.cc | 70 bool allow_small_batch, in TryDequeueMany() argument
100 [callback, allow_small_batch, in TryDequeueMany()
124 if (allow_small_batch && !queues_[0].empty()) { in TryDequeueMany()
130 if (allow_small_batch) { in TryDequeueMany()
|
| D | padding_fifo_queue.h | 46 bool allow_small_batch,
|
| D | fifo_queue.h | 48 bool allow_small_batch,
|
| D | priority_queue.h | 65 bool allow_small_batch,
|
| D | barrier_ops.cc | 201 void TryTakeMany(int num_elements, bool allow_small_batch, int64 timeout, in TryTakeMany() argument
209 if (allow_small_batch) { in TryTakeMany()
234 num_elements_to_deliver, ctx, allow_small_batch, in TryTakeMany()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | queue_interface.h | 63 bool allow_small_batch,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BarrierTakeMany.pbtxt | 44 name: "allow_small_batch"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 1035 allow_small_batch=False, argument
1081 allow_small_batch,
1089 if allow_small_batch:
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | barrier_ops_test.py | 154 index_t, key_t, value_list_t = b.take_many(3, allow_small_batch=True)
505 take_t = b.take_many(1, allow_small_batch=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 273 …argspec: "args=[\'handle\', \'num_elements\', \'component_types\', \'allow_small_batch\', \'wait_f…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 273 …argspec: "args=[\'handle\', \'num_elements\', \'component_types\', \'allow_small_batch\', \'wait_f…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 4340 name: "allow_small_batch"
|
| D | ops_history.v2.pbtxt | 8747 name: "allow_small_batch"
|
| D | ops_history.v1.pbtxt | 8747 name: "allow_small_batch"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 2922 name: "allow_small_batch"
|