Searched refs:scalar_cache (Results 1 – 3 of 3) sorted by relevance
123 scalar_cache = ctx.scalar_cache()124 tensor = scalar_cache.get(cache_key, None)128 scalar_cache[cache_key] = t
467 context.get_default_context().scalar_cache().clear()
70 self.scalar_cache = {}251 def scalar_cache(self): member in Context253 return self._eager_context.scalar_cache