| /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,
276 bool allow_small_batch, in TryDequeueMany() argument
338 [callback, allow_small_batch, in TryDequeueMany()
364 if (allow_small_batch && !queues_[0].empty()) { in TryDequeueMany()
370 if (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 | priority_queue.cc | 240 bool allow_small_batch, in TryDequeueMany() argument
302 [callback, this, allow_small_batch]( in TryDequeueMany()
311 (!allow_small_batch && s < attempt->elements_requested)) { in TryDequeueMany()
335 if (s == 0 || !(closed_ && allow_small_batch)) { in TryDequeueMany()
|
| D | padding_fifo_queue.h | 46 bool allow_small_batch,
|
| D | priority_queue.h | 65 bool allow_small_batch,
|
| D | fifo_queue.h | 48 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/ops/compat/ops_history_v1/ |
| D | BarrierTakeMany.pbtxt | 31 name: "allow_small_batch"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BarrierTakeMany.pbtxt | 31 name: "allow_small_batch"
|
| /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 | 1113 allow_small_batch=False, argument
1159 allow_small_batch,
1167 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)
507 take_t = b.take_many(1, allow_small_batch=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 313 …argspec: "args=[\'handle\', \'num_elements\', \'component_types\', \'allow_small_batch\', \'wait_f…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 313 …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"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3270 name: "allow_small_batch"
|