Searched refs:ManyOutShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fifo_queue.cc | 235 ManyOutShape(i, 0), &element); in TryDequeueMany()
318 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
|
| D | queue_base.h | 115 TensorShape ManyOutShape(int i, int64 batch_size) { in ManyOutShape() function
|
| D | priority_queue.cc | 279 ManyOutShape(i, 0), &element); in TryDequeueMany()
349 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 314 Status s = ctx->allocate_temp(component_dtypes_[i], ManyOutShape(i, 0), in TryDequeueMany()
397 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
|
| D | queue_base.cc | 167 const TensorShape expected_shape = ManyOutShape(i, batch_size); in ValidateManyTuple()
|
| D | padding_fifo_queue.cc | 82 ManyOutShape(i, 0), &element)); in TryDequeueMany()
|