Searched refs:zeros_cache (Results 1 – 2 of 2) sorted by relevance
187 def zeros_cache(self): member in _TensorCaches545 self.zeros_cache().flush()833 def zeros_cache(self): member in Context835 return _tensor_caches_map[self._id].zeros_cache
683 cached = ctx.zeros_cache().get(cache_key)690 ctx.zeros_cache().put(cache_key, cached)