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