Home
last modified time | relevance | path

Searched +path:aten +path:native +path:vulkan (Results 1 – 25 of 216) sorted by relevance

123456789

/external/pytorch/aten/src/ATen/native/vulkan/
DVulkanGuardImpl.cpp
DVulkanOpaqueTensorImpl.h
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DFlip.cpp
DNativeLayerNorm.cpp
DLayernorm.h
DMean.cpp
/external/pytorch/aten/src/ATen/native/vulkan/glsl/
Dquantized_upsample_nearest2d.glsl
Dselect_height_3d.glsl
Dconv2d_dw_output_tile.glsl
Dtanh.glsl
Dupsample_nearest2d.glsl
Dtexel_access.h
Dcumsum_channel.glsl
Duniform_.glsl
Dconvert_channels_to_height_packed.glsl
Dconv1d.glsl
Dquantized_gelu_tanh_qint8.glsl
Dconv2d_pw.glsl
Dupsample_bilinear2d_align_false.glsl
Dquantize_per_tensor_qint32.glsl
/external/pytorch/aten/src/ATen/native/vulkan/api/
DTensor.cpp
DPipeline.cpp
DQueryPool.h
DAllocator.cpp
DResource.h

123456789