Searched refs:max_tensor_size (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | simple_memory_arena_debug_dump.cc | 145 size_t max_tensor_size = 0; in DumpArenaInfo() local
152 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()
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | distributed_c10d.py | 3241 max_tensor_size = max(tensor_sizes) # type: ignore[possibly-undefined]
3243 tensor.resize_(max_tensor_size)
3245 max_tensor_size = torch.tensor([0], dtype=torch.long, device=pg_device)
3246 broadcast(max_tensor_size, src=src, group=group)
3250 max_tensor_size.item(), dtype=torch.uint8, device=pg_device
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | context.py | 123 def __init__(self, max_items=256, max_tensor_size=10000): argument
126 self._max_tensor_size = max_tensor_size
|
| H A D | core_test.py | 1142 cache = context._EagerTensorCache(max_items=100, max_tensor_size=3)
|