Home
last modified time | relevance | path

Searched refs:local_tpu_cache_tensor (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py1559 def merge_caches_on_tpu(self, local_tpu_cache_tensor): argument
1571 local_tpu_cache_tensor,
1573 local_tpu_cache_tensor.shape.as_list())