Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dmemory_planner.cpp59 size_t compute_aligned_tensor_size(size_t nbytes) { in compute_aligned_tensor_size() function
361 compute_aligned_tensor_size(tensor->storage().nbytes()); in deallocateOutputTensors()
460 size_t current_size = compute_aligned_tensor_size(storage.nbytes()); in deallocateManagedTensors()