Searched refs:zeros_cache (Results 1 – 2 of 2) sorted by relevance
691 cached = ctx.zeros_cache().get(cache_key)698 ctx.zeros_cache().put(cache_key, cached)
219 def zeros_cache(self): member in _TensorCaches618 self.zeros_cache().flush()958 def zeros_cache(self): member in Context960 return _tensor_caches_map[self._id].zeros_cache