Searched refs:ones_rank_cache (Results 1 – 2 of 2) sorted by relevance
155 def ones_rank_cache(self): member in _ThreadLocalData365 self.ones_rank_cache().flush()482 def ones_rank_cache(self): member in Context484 return self._thread_local_data.ones_rank_cache
67 new_shape = ctx.ones_rank_cache().get(rank)70 ctx.ones_rank_cache().put(rank, new_shape)