Home
last modified time | relevance | path

Searched defs:allow_small_batch (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfifo_queue.cc196 bool allow_small_batch, in TryDequeueMany()
H A Dpriority_queue.cc238 bool allow_small_batch, in TryDequeueMany()
H A Dpadding_fifo_queue.cc70 bool allow_small_batch, in TryDequeueMany()
H A Drandom_shuffle_queue_op.cc278 bool allow_small_batch, in TryDequeueMany()
H A Dbarrier_ops.cc202 void TryTakeMany(int num_elements, bool allow_small_batch, int64_t timeout, in TryTakeMany()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Ddata_flow_ops.py1114 allow_small_batch=False, argument