Searched refs:_tensor_caches_map (Results 1 – 1 of 1) sorted by relevance
167 _tensor_caches_map = {} variable339 if _tensor_caches_map is None:341 if self._context_id in _tensor_caches_map:342 del _tensor_caches_map[self._context_id]399 _tensor_caches_map[self._id] = _TensorCaches()831 return _tensor_caches_map[self._id].ones_rank_cache835 return _tensor_caches_map[self._id].zeros_cache