Searched refs:tpu_host_placement_function (Results 1 – 2 of 2) sorted by relevance
159 def tpu_host_placement_function(self): member in TPUContext168 return self._internal_ctx.tpu_host_placement_function(host_id=host_id)522 def tpu_host_placement_function(self): member in _InternalTPUContext
951 device_0 = ctx.tpu_host_placement_function(host_id=0)980 return ctx.tpu_host_placement_function(replica_id=replica_id)990 with ops.device(ctx.tpu_host_placement_function(host_id=host_id)):1240 tpu_host_placement_fn = self._ctx.tpu_host_placement_function1899 with ops.device(self._ctx.tpu_host_placement_function(replica_id=0)):