Searched refs:_handle_graph (Results 1 – 3 of 3) sorted by relevance
42 def _handle_graph(handle): function188 with _handle_graph(self.handle), self._assign_dependencies():198 with _handle_graph(self.handle), self._assign_dependencies():209 with _handle_graph(self.handle), self._assign_dependencies():
215 def _handle_graph(handle): function270 with _handle_graph(handle):993 with _handle_graph(self.handle), self._assign_dependencies():1017 with _handle_graph(self.handle), self._assign_dependencies():1052 with _handle_graph(self.handle):1346 with _handle_graph(self.handle), self._assign_dependencies():
973 def _handle_graph(self, handle): member in TPUMirroredVariable1006 with self._handle_graph(var.handle):1021 with self._handle_graph(var.handle):1036 with self._handle_graph(var.handle):