Searched refs:max_tensor_size (Results 1 – 2 of 2) sorted by relevance
90 def __init__(self, max_items=256, max_tensor_size=10000): argument93 self._max_tensor_size = max_tensor_size
1131 cache = context._EagerTensorCache(max_items=100, max_tensor_size=3)