Home
last modified time | relevance | path

Searched defs:gpu_sizes (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/
H A DTensor.cpp98 std::vector<int64_t> gpu_sizes; in calc_gpu_sizes() local
154 const std::vector<int64_t>& gpu_sizes, in create_image_extents()
532 const std::vector<int64_t>& gpu_sizes, in vTensorStorage()
655 const std::vector<int64_t>& gpu_sizes, in discard_and_reallocate()
H A DTensor.h272 inline const std::vector<int64_t>& gpu_sizes() const { in gpu_sizes() function
/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dvulkan_compute_api_test.cpp2942 const std::vector<int64_t>& gpu_sizes, in test_conv2d()