Searched refs:local_tpu_cache_tensor (Results 1 – 1 of 1) sorted by relevance
1663 def merge_caches_on_tpu(self, local_tpu_cache_tensor): argument1673 local_tpu_cache_tensor,1675 local_tpu_cache_tensor.shape.as_list())