Searched refs:max_tensor_size (Results 1 – 3 of 3) sorted by relevance
145 size_t max_tensor_size = 0; in DumpArenaInfo() local152 if (alloc_info.size > max_tensor_size) { in DumpArenaInfo()155 max_tensor_size = alloc_info.size; in DumpArenaInfo()156 } else if (alloc_info.size == max_tensor_size) { in DumpArenaInfo()162 max_size_tensors.size(), max_tensor_size, in DumpArenaInfo()163 static_cast<float>(max_tensor_size) / (1 << 20)); in DumpArenaInfo()
123 def __init__(self, max_items=256, max_tensor_size=10000): argument126 self._max_tensor_size = max_tensor_size
1142 cache = context._EagerTensorCache(max_items=100, max_tensor_size=3)