Google Git
Sign in
android / platform / external / pytorch / ac3190c52c / . / aten / src / ATen / native / vulkan / glsl
tree: c6d943a588b9db15cfe5f83ff54de817c060fbe0 [path history] [tgz]
  1. templates/
  2. abs.glsl
  3. abs_.glsl
  4. adaptive_avg_pool2d.glsl
  5. addmm.glsl
  6. avg_pool2d.glsl
  7. baddbmm.glsl
  8. batchnorm.glsl
  9. bmm.glsl
  10. buffer_to_buffer.glsl
  11. cat_feature.glsl
  12. clamp.glsl
  13. clamp_.glsl
  14. conv2d.glsl
  15. conv2d_dw.glsl
  16. conv_transpose2d.glsl
  17. cumsum.glsl
  18. dequantize.glsl
  19. flip.glsl
  20. gelu_tanh.glsl
  21. gelu_tanh_.glsl
  22. glu_channel.glsl
  23. glu_channel_mul4.glsl
  24. hardshrink.glsl
  25. hardshrink_.glsl
  26. hardsigmoid.glsl
  27. hardsigmoid_.glsl
  28. hardswish.glsl
  29. hardswish_.glsl
  30. image2d_to_nchw.glsl
  31. image_to_nchw.glsl
  32. image_to_nchw_int32.glsl
  33. image_to_nchw_quantized_mul4.glsl
  34. image_to_nchw_uint.glsl
  35. indexing.h
  36. layernorm.glsl
  37. leaky_relu.glsl
  38. leaky_relu_.glsl
  39. lerp.glsl
  40. lerp_.glsl
  41. lerp_scalar.glsl
  42. lerp_scalar_.glsl
  43. log_softmax.glsl
  44. masked_fill.glsl
  45. max_pool2d.glsl
  46. mean.glsl
  47. mean2d.glsl
  48. mm.glsl
  49. nchw_to_image.glsl
  50. nchw_to_image2d.glsl
  51. nchw_to_image_bool.glsl
  52. nchw_to_image_int32.glsl
  53. nchw_to_image_int8.glsl
  54. nchw_to_image_uint8.glsl
  55. normal_.glsl
  56. permute_4d.glsl
  57. pow_scalar_tensor.glsl
  58. quantize_per_tensor_qint32.glsl
  59. quantize_per_tensor_qint8.glsl
  60. quantize_per_tensor_quint8.glsl
  61. quantized_add.glsl
  62. quantized_conv2d.glsl
  63. quantized_conv2d_dw.glsl
  64. quantized_conv2d_pw_2x2.glsl
  65. quantized_div.glsl
  66. quantized_mul.glsl
  67. quantized_sub.glsl
  68. quantized_upsample_nearest2d.glsl
  69. random.h
  70. reflection_pad2d.glsl
  71. replication_pad2d.glsl
  72. select_batch_4d.glsl
  73. select_depth_3d.glsl
  74. select_depth_4d.glsl
  75. select_height_3d.glsl
  76. select_height_4d.glsl
  77. select_width_3d.glsl
  78. select_width_4d.glsl
  79. sigmoid.glsl
  80. sigmoid_.glsl
  81. slice_4d.glsl
  82. softmax_batch_height_width.glsl
  83. softmax_channel.glsl
  84. stack_feature.glsl
  85. sum_dim.glsl
  86. sum_dim_keepdim.glsl
  87. tanh.glsl
  88. tanh_.glsl
  89. texel_access.h
  90. threshold.glsl
  91. uniform_.glsl
  92. unsqueeze.glsl
  93. upsample_bilinear2d_align_false.glsl
  94. upsample_bilinear2d_align_true.glsl
  95. upsample_nearest2d.glsl
  96. zero.glsl