Home
last modified time | relevance | path

Searched defs:gpu_memory_layout (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/
H A DTensor.cpp531 const api::GPUMemoryLayout gpu_memory_layout, in vTensorStorage()
656 const api::GPUMemoryLayout gpu_memory_layout, in discard_and_reallocate()
H A DTensor.h256 inline api::GPUMemoryLayout gpu_memory_layout() const { in gpu_memory_layout() function