(self)
| 546 | return self._values[replica_id] |
| 547 | |
| 548 | def _as_graph_element(self): |
| 549 | obj = self.get() |
| 550 | conv_fn = getattr(obj, "_as_graph_element", None) |
| 551 | if conv_fn and callable(conv_fn): |
| 552 | return conv_fn() |
| 553 | return obj |
| 554 | |
| 555 | |
| 556 | def _assign_on_device(device, variable, tensor): |
no test coverage detected