Searched refs:segment_pool_id (Results 1 – 2 of 2) sorted by relevance
207 IValue segment_pool_id = "segment_pool_id"; in _memory_snapshot_pickled() local255 segment_pool_id, in _memory_snapshot_pickled()
666 py::str segment_pool_id = "segment_pool_id"; in THCPModule_memorySnapshot() local708 segmentDict[segment_pool_id] = segmentInfo.owner_private_pool_id; in THCPModule_memorySnapshot()