Select which examples to enqueue based on vector `keep_input`.
(tensor_list, keep_input)
| 710 | |
| 711 | |
| 712 | def _select_which_to_enqueue(tensor_list, keep_input): |
| 713 | """Select which examples to enqueue based on vector `keep_input`.""" |
| 714 | select_i = math_ops.cast(keep_input, dtypes.int32) |
| 715 | tensor_list = [ |
| 716 | data_flow_ops.dynamic_partition(x, select_i, num_partitions=2)[1] |
| 717 | for x in tensor_list] |
| 718 | return tensor_list |
| 719 | |
| 720 | |
| 721 | def _enqueue_join(queue, tensor_list_list, enqueue_many, keep_input): |
no test coverage detected