Directory listing for /PyTorch/pytorch/aten/src/ATen/native/vulkan/glsl/
abs.glsl
abs_.glsl
adaptive_avg_pool2d.glsl
avg_pool2d.glsl
batchnorm.glsl
binary_op_scalar.glsl
binary_op_tensor.glsl
buffer_to_buffer.glsl
cat_feature.glsl
clamp.glsl
clamp_.glsl
conv1d.glsl
conv2d.glsl
conv2d_dw.glsl
conv2d_dw_output_tile.glsl
conv2d_pw.glsl
conv_transpose2d.glsl
convert_channels_to_height_packed.glsl
convert_channels_to_width_packed.glsl
cumsum_batch_height_width.glsl
cumsum_channel.glsl
dequantize.glsl
extract_texel.glsl
flip.glsl
gelu_tanh.glsl
gelu_tanh_.glsl
glu_channel.glsl
glu_channel_mul4.glsl
hardshrink.glsl
hardshrink_.glsl
hardsigmoid.glsl
hardsigmoid_.glsl
hardswish.glsl
hardswish_.glsl
image2d_to_nchw.glsl
image_to_nchw.glsl
image_to_nchw_int32.glsl
image_to_nchw_quantized_mul4.glsl
image_to_nchw_uint.glsl
indexing.h
leaky_relu.glsl
leaky_relu_.glsl
lerp.glsl
lerp_.glsl
lerp_scalar.glsl
lerp_scalar_.glsl
log_softmax.glsl
masked_fill.glsl
max_pool2d.glsl
mean_dim.glsl
mean_dim_keepdim.glsl
mm.glsl
nchw_to_image.glsl
nchw_to_image2d.glsl
nchw_to_image2d_int32.glsl
nchw_to_image2d_int8.glsl
nchw_to_image2d_uint8.glsl
nchw_to_image_bool.glsl
nchw_to_image_int32.glsl
nchw_to_image_int8.glsl
nchw_to_image_uint8.glsl
normal_.glsl
permute_4d.glsl
pow_scalar_tensor.glsl
quantize_per_tensor_qint32.glsl
quantize_per_tensor_qint8.glsl
quantize_per_tensor_quint8.glsl
quantized_add.glsl
quantized_addmm_qint8.glsl
quantized_addmm_quint8.glsl
quantized_clamp_qint8.glsl
quantized_clamp_qint8_.glsl
quantized_clamp_quint8.glsl
quantized_clamp_quint8_.glsl
quantized_conv2d.glsl
quantized_conv2d_dw.glsl
quantized_conv2d_pw_2x2.glsl
quantized_conv_transpose2d.glsl
quantized_div.glsl
quantized_gelu_tanh_qint8.glsl
quantized_gelu_tanh_qint8_.glsl
quantized_gelu_tanh_quint8.glsl
quantized_gelu_tanh_quint8_.glsl
quantized_max_pool2d_qint8.glsl
quantized_max_pool2d_quint8.glsl
quantized_mm_qint8.glsl
quantized_mm_quint8.glsl
quantized_mul.glsl
quantized_sub.glsl
quantized_upsample_nearest2d.glsl
random.h
reflection_pad2d.glsl
replication_pad2d.glsl
select_batch_4d.glsl
select_depth_3d.glsl
select_depth_4d.glsl
select_height_3d.glsl
select_height_4d.glsl
select_width_3d.glsl
select_width_4d.glsl
shader_params.yaml
sigmoid.glsl
sigmoid_.glsl
slice_4d.glsl
softmax_batch_height_width.glsl
softmax_channel.glsl
stack_feature.glsl
sum_dim.glsl
sum_dim_keepdim.glsl
tanh.glsl
tanh_.glsl
texel_access.h
threshold.glsl
unary_op.glsl
uniform_.glsl
unsqueeze.glsl
upsample_bilinear2d_align_false.glsl
upsample_bilinear2d_align_true.glsl
upsample_nearest2d.glsl
zero.glsl