Home
last modified time | relevance | path

Searched +path:pytorch +path:aten +path:src +path:aten +path:vulkan (Results 1 – 25 of 218) sorted by relevance

123456789

/aosp_15_r20/external/pytorch/aten/src/ATen/vulkan/
H A DContext.h
H A DContext.cpp
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/
H A DVulkanOpaqueTensorImpl.h
H A DVulkanGuardImpl.cpp
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DStack.cpp
H A DClone.cpp
H A DSoftmax.cpp
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/glsl/
H A Dselect_width_3d.glsl
H A Dselect_depth_3d.glsl
H A Dgelu_tanh.glsl
H A Dadaptive_avg_pool2d.glsl
H A Dnchw_to_image2d.glsl
H A Dquantize_per_tensor_qint32.glsl
H A Dquantized_addmm_quint8.glsl
H A Dquantized_gelu_tanh_quint8.glsl
H A Dimage2d_to_nchw.glsl
H A Dzero.glsl
H A Dextract_texel.glsl
H A Dabs.glsl
H A Dsoftmax_batch_height_width.glsl
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/
H A DShaderRegistry.cpp
H A DShaderRegistry.h
H A DResource.cpp
H A DDescriptor.cpp
H A DPipeline.cpp

123456789