Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/inductor/
H A Dinductor_ops.cpp36 Tensor _alloc_from_pool( in _alloc_from_pool() function
100 _alloc_from_pool, in TORCH_LIBRARY_FRAGMENT()
H A Dinductor_ops.h23 TORCH_API at::Tensor _alloc_from_pool(
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_common.cpp1172 *ret_new_tensor = new_tensor_handle(torch::inductor::_alloc_from_pool( in aoti_torch__alloc_from_pool()