Searched refs:scalar_cache (Results 1 – 2 of 2) sorted by relevance
105 scalar_cache = ctx.scalar_cache()106 tensor = scalar_cache.get(cache_key, None)111 scalar_cache[cache_key] = t
148 self.scalar_cache = {}364 self.scalar_cache().clear()478 def scalar_cache(self): member in Context480 return self._thread_local_data.scalar_cache