Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dmemory_snapshot.cpp295 IValue segment_alloc_s = "segment_alloc"; in _memory_snapshot_pickled() local
314 return segment_alloc_s; in _memory_snapshot_pickled()
H A DModule.cpp745 py::str segment_alloc_s = "segment_alloc"; in THCPModule_memorySnapshot() local
765 return segment_alloc_s; in THCPModule_memorySnapshot()