Take work from local worker.
()
| 380 | return local_work |
| 381 | |
| 382 | def local_take(): |
| 383 | """Take work from local worker.""" |
| 384 | assert self._local_work_mgr, 'local_work_mgr should not be None.' |
| 385 | return self._local_work_mgr.take() |
| 386 | |
| 387 | if self._local_work_mgr: |
| 388 | with ops.name_scope(self.name): |