Searched refs:_EagerTensorCache (Results 1 – 2 of 2) sorted by relevance
60 class _EagerTensorCache(object): class157 self._ones_rank_cache = _EagerTensorCache()163 self._zeros_cache = _EagerTensorCache()
763 cache = context._EagerTensorCache(max_items=100, max_tensor_size=3)