Searched refs:gpu_memory_layout (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Mm.cpp | 38 if (v_input.gpu_memory_layout() == in pack_inputs_using_width_packing()
44 v_input.gpu_memory_layout() == api::GPUMemoryLayout::TENSOR_WIDTH_PACKED, in pack_inputs_using_width_packing()
70 if (v_weight.gpu_memory_layout() == in pack_weights_using_height_packing()
77 v_weight.gpu_memory_layout() == in pack_weights_using_height_packing()
684 v_input.gpu_memory_layout() == api::GPUMemoryLayout::TENSOR_WIDTH_PACKED, in run_addmm_context()
688 packed_v_weight.gpu_memory_layout() == in run_addmm_context()
794 packed_v_input.gpu_memory_layout() == in run_baddbmm_context()
799 packed_v_weight.gpu_memory_layout() == in run_baddbmm_context()
|
| H A D | Copy.cpp | 299 switch (v_src.gpu_memory_layout()) { in from_vulkan()
|
| H A D | Convolution.cpp | 804 if (v_weight.gpu_memory_layout() == in pack_weights_using_width_packing()
810 v_weight.gpu_memory_layout() == api::GPUMemoryLayout::TENSOR_WIDTH_PACKED, in pack_weights_using_width_packing()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/ |
| H A D | Tensor.h | 36 const api::GPUMemoryLayout gpu_memory_layout,
88 const api::GPUMemoryLayout gpu_memory_layout,
256 inline api::GPUMemoryLayout gpu_memory_layout() const { in gpu_memory_layout() function
|
| H A D | Tensor.cpp | 531 const api::GPUMemoryLayout gpu_memory_layout, in vTensorStorage() argument
538 create_image_extents(gpu_sizes, storage_type, gpu_memory_layout)),
656 const api::GPUMemoryLayout gpu_memory_layout, in discard_and_reallocate() argument
663 extents_ = create_image_extents(gpu_sizes, storage_type_, gpu_memory_layout); in discard_and_reallocate()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 3616 v_output.gpu_memory_layout(), api::GPUMemoryLayout::TENSOR_HEIGHT_PACKED); in TEST_F()
3668 v_output.gpu_memory_layout(), api::GPUMemoryLayout::TENSOR_WIDTH_PACKED); in TEST_F()
|