Home
last modified time | relevance | path

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

/external/pytorch/c10/core/
DCachingDeviceAllocator.h48 using StatTypes = std::array<bool, static_cast<size_t>(StatType::NUM_TYPES)>; variable
51 void for_each_selected_stat_type(const StatTypes& stat_types, Func f) { in for_each_selected_stat_type()
/external/pytorch/c10/xpu/
DXPUCachingAllocator.cpp122 StatTypes stat_types = {};
183 StatTypes stat_types = get_stat_types_for_pool(pool); in free_block()
300 StatTypes stat_types = get_stat_types_for_pool(*pool); in release_block()
338 StatTypes get_stat_types_for_pool(const BlockPool& pool) { in get_stat_types_for_pool()
339 StatTypes stat_types = {}; in get_stat_types_for_pool()
457 StatTypes stat_types = get_stat_types_for_pool(*block->pool); in free()
/external/pytorch/c10/cuda/
DCUDACachingAllocator.cpp141 const StatTypes& stat_types) { in decrease_stat_array()
756 StatTypes stat_types = {false};
1461 StatTypes stat_types = get_stat_types_for_pool(*block->pool); in free()
2256 StatTypes stat_types = get_stat_types_for_pool(*to_map->pool); in map_block()
2359 StatTypes stat_types = get_stat_types_for_pool(pool); in free_block()
2450 StatTypes get_stat_types_for_pool(const BlockPool& pool) { in get_stat_types_for_pool()
2451 StatTypes stat_types = {false}; in get_stat_types_for_pool()
2840 StatTypes stat_types = get_stat_types_for_pool(*pool); in release_block()
2902 StatTypes stat_types = get_stat_types_for_pool(*block->pool); in unmap_block()