Searched refs:_TensorCaches (Results 1 – 1 of 1) sorted by relevance
170 class _TensorCaches(threading.local): class176 super(_TensorCaches, self).__init__()399 _tensor_caches_map[self._id] = _TensorCaches()