Home
last modified time | relevance | path

Searched refs:from_allocation (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/profiler/
D_memory_profiler.py138 def from_allocation(cls, alloc: _ExtraFields_Allocation) -> Optional["TensorKey"]: member in TensorKey
367 key = TensorKey.from_allocation(alloc_fields)
470 key = TensorKey.from_allocation(i.typed[1])
480 edges[TensorKey.from_allocation(i.typed[1])].input_version = None
690 tkey = TensorKey.from_allocation(alloc_fields)
/external/pytorch/torch/cuda/
D_memory_viz.py430 key = TensorKey.from_allocation(event.extra_fields)