Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dbackprop.py691 cached = ctx.zeros_cache().get(cache_key)
698 ctx.zeros_cache().put(cache_key, cached)
Dcontext.py219 def zeros_cache(self): member in _TensorCaches
618 self.zeros_cache().flush()
958 def zeros_cache(self): member in Context
960 return _tensor_caches_map[self._id].zeros_cache