Enqueue `tensor_list_list` in `queue`.
(queue, tensor_list_list, enqueue_many, keep_input)
| 719 | |
| 720 | |
| 721 | def _enqueue_join(queue, tensor_list_list, enqueue_many, keep_input): |
| 722 | """Enqueue `tensor_list_list` in `queue`.""" |
| 723 | if enqueue_many: |
| 724 | enqueue_fn = queue.enqueue_many |
| 725 | else: |
| 726 | enqueue_fn = queue.enqueue |
| 727 | if keep_input.shape.ndims == 1: |
| 728 | enqueue_ops = [enqueue_fn(_select_which_to_enqueue(x, keep_input)) |
| 729 | for x in tensor_list_list] |
| 730 | else: |
| 731 | enqueue_ops = [utils.smart_cond( |
| 732 | keep_input, |
| 733 | lambda: enqueue_fn(tl), # pylint:disable=cell-var-from-loop |
| 734 | control_flow_ops.no_op) for tl in tensor_list_list] |
| 735 | queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops)) |
| 736 | |
| 737 | |
| 738 | def _enqueue(queue, tensor_list, threads, enqueue_many, keep_input): |
no test coverage detected