Deals with doing gather differently for resource variables.
(params, ids, name=None)
| 807 | |
| 808 | |
| 809 | def _do_gather(params, ids, name=None): |
| 810 | """Deals with doing gather differently for resource variables.""" |
| 811 | if isinstance(params, resource_variable_ops.ResourceVariable): |
| 812 | return params.sparse_read(ids, name=name) |
| 813 | return array_ops.gather(params, ids, name=name) |
| 814 | |
| 815 | |
| 816 | def _embedding_lookup_with_distributed_aggregation(params, |
no test coverage detected