Home
last modified time | relevance | path

Searched refs:allocator_settings (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/cuda/
Dmemory_snapshot.cpp361 auto allocator_settings = new_dict(); in _memory_snapshot_pickled() local
371 allocator_settings.insert( in _memory_snapshot_pickled()
374 allocator_settings.insert( in _memory_snapshot_pickled()
376 allocator_settings.insert( in _memory_snapshot_pickled()
379 allocator_settings.insert( in _memory_snapshot_pickled()
381 allocator_settings.insert( in _memory_snapshot_pickled()
384 allocator_settings.insert( in _memory_snapshot_pickled()
387 allocator_settings.insert( in _memory_snapshot_pickled()
397 allocator_settings.insert(roundup_power2_divisions_s, roundup_settings); in _memory_snapshot_pickled()
402 result.insert("allocator_settings", allocator_settings); in _memory_snapshot_pickled()
DModule.cpp812 py::dict allocator_settings; in THCPModule_memorySnapshot() local
822 allocator_settings[last_allocator_settings_s] = in THCPModule_memorySnapshot()
824 allocator_settings[max_split_size_s] = in THCPModule_memorySnapshot()
826 allocator_settings[garbage_collection_threshold_s] = in THCPModule_memorySnapshot()
828 allocator_settings[expandable_segments_s] = in THCPModule_memorySnapshot()
830 allocator_settings[pinned_num_register_threads_s] = in THCPModule_memorySnapshot()
832 allocator_settings[release_lock_on_malloc_s] = in THCPModule_memorySnapshot()
834 allocator_settings[pinned_use_host_register_s] = in THCPModule_memorySnapshot()
843 allocator_settings[roundup_power2_divisions_s] = roundup_settings; in THCPModule_memorySnapshot()
848 result["allocator_settings"] = allocator_settings; in THCPModule_memorySnapshot()
/external/pytorch/torch/utils/viz/
DMemoryViz.js1288 .text(JSON.stringify(snapshot.allocator_settings, null, 2));