Enqueue `tensor_list` in `queue`.
(queue, tensor_list, threads, enqueue_many, keep_input)
| 736 | |
| 737 | |
| 738 | def _enqueue(queue, tensor_list, threads, enqueue_many, keep_input): |
| 739 | """Enqueue `tensor_list` in `queue`.""" |
| 740 | if enqueue_many: |
| 741 | enqueue_fn = queue.enqueue_many |
| 742 | else: |
| 743 | enqueue_fn = queue.enqueue |
| 744 | if keep_input.shape.ndims == 1: |
| 745 | enqueue_ops = [ |
| 746 | enqueue_fn(_select_which_to_enqueue(tensor_list, keep_input))] * threads |
| 747 | else: |
| 748 | enqueue_ops = [utils.smart_cond( |
| 749 | keep_input, |
| 750 | lambda: enqueue_fn(tensor_list), |
| 751 | control_flow_ops.no_op)] * threads |
| 752 | queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops)) |
| 753 | |
| 754 | |
| 755 | def _which_queue(dynamic_pad): |
no test coverage detected