Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dmemory_snapshot.cpp207 IValue segment_pool_id = "segment_pool_id"; in _memory_snapshot_pickled() local
255 segment_pool_id, in _memory_snapshot_pickled()
H A DModule.cpp666 py::str segment_pool_id = "segment_pool_id"; in THCPModule_memorySnapshot() local
708 segmentDict[segment_pool_id] = segmentInfo.owner_private_pool_id; in THCPModule_memorySnapshot()