Google Git
Sign in
android / platform / external / pytorch / 4c70ab26ef0a596d779ec38e9bc9c1bfc4373f93 / . / aten / src / ATen / native / cuda
tree: d69d8e9958ff7e3790521cb59836db46848fb4ec [path history] [tgz]
  1. cutlass_extensions/
  2. linalg/
  3. AbsKernel.cu
  4. Activation.cpp
  5. Activation.h
  6. ActivationEluKernel.cu
  7. ActivationGeluKernel.cu
  8. ActivationGluKernel.cu
  9. ActivationHardshrinkKernel.cu
  10. ActivationHardsigmoidKernel.cu
  11. ActivationHardswishKernel.cu
  12. ActivationHardtanhKernel.cu
  13. ActivationLeakyReluKernel.cu
  14. ActivationLogSigmoidKernel.cu
  15. ActivationMishKernel.cu
  16. ActivationPreluKernel.cu
  17. ActivationSiluKernel.cu
  18. ActivationSoftplusKernel.cu
  19. ActivationSoftshrinkKernel.cu
  20. ActivationThresholdKernel.cu
  21. AdaptiveAveragePooling.cu
  22. AdaptiveAveragePooling3d.cu
  23. AdaptiveMaxPooling2d.cu
  24. AdaptiveMaxPooling3d.cu
  25. airy_ai.cu
  26. AmpKernels.cu
  27. AveragePool2d.cu
  28. AveragePool3d.cu
  29. bessel_j0.cu
  30. bessel_j1.cu
  31. bessel_y0.cu
  32. bessel_y1.cu
  33. BinaryBitwiseOpsKernels.cu
  34. BinaryDivFloorKernel.cu
  35. BinaryDivTrueKernel.cu
  36. BinaryDivTruncKernel.cu
  37. BinaryGeometricKernels.cu
  38. BinaryInternal.h
  39. BinaryLogicalOpsKernels.cu
  40. BinaryMiscBackwardOpsKernels.cu
  41. BinaryMiscOpsKernels.cu
  42. BinaryMulKernel.cu
  43. BinaryRemainderKernel.cu
  44. BinaryShiftOpsKernels.cu
  45. Blas.cpp
  46. block_reduce.cuh
  47. Bucketization.cu
  48. chebyshev_polynomial_t.cu
  49. chebyshev_polynomial_u.cu
  50. chebyshev_polynomial_v.cu
  51. chebyshev_polynomial_w.cu
  52. Col2Im.cu
  53. CompareEQKernel.cu
  54. CompareKernels.cu
  55. ComplexKernel.cu
  56. CompositeRandomAccessor.h
  57. ConvolutionMM2d.cu
  58. Copy.cu
  59. Copy.h
  60. CopysignKernel.cu
  61. CrossKernel.cu
  62. CUDAJitLoops.cuh
  63. CUDALoops.cuh
  64. CUDAScalar.cu
  65. CuFFTPlanCache.h
  66. CuFFTUtils.h
  67. CumminmaxKernel.cu
  68. CumprodKernel.cu
  69. CumsumKernel.cu
  70. DepthwiseConv2d.cu
  71. DepthwiseConv3d.cu
  72. DeviceSqrt.cuh
  73. DilatedMaxPool2d.cu
  74. DilatedMaxPool3d.cu
  75. DistanceKernel.cu
  76. DistributionBernoulli.cu
  77. DistributionCauchyKernel.cu
  78. DistributionExponentialKernel.cu
  79. DistributionGeometricKernel.cu
  80. DistributionLogNormalKernel.cu
  81. DistributionNormal.cu
  82. DistributionRandomKernel.cu
  83. Distributions.cpp
  84. Distributions.cu
  85. Distributions.h
  86. DistributionTemplates.h
  87. DistributionUniform.cu
  88. Dropout.cu
  89. Embedding.cu
  90. EmbeddingBackwardKernel.cu
  91. EmbeddingBackwardKernel.cuh
  92. EmbeddingBag.cu
  93. Equal.cpp
  94. FillKernel.cu
  95. FlattenIndicesKernel.cu
  96. ForeachBinaryOpList.cu
  97. ForeachBinaryOpScalar.cu
  98. ForeachBinaryOpScalarList.cu
  99. ForeachBinaryOpScalarTensor.cu
  100. ForeachFunctors.cuh
  101. ForeachMinMaxFunctors.cuh
  102. ForeachPointwiseOp.cu
  103. ForeachReduceOp.cu
  104. ForeachTernaryOp.cu
  105. ForeachUnaryOp.cu
  106. FractionalMaxPool2d.cu
  107. FractionalMaxPool3d.cu
  108. FunctionOfAMatrixUtilsKernel.cu
  109. fused_adam_amsgrad_impl.cu
  110. fused_adam_amsgrad_impl.cuh
  111. fused_adam_impl.cu
  112. fused_adam_impl.cuh
  113. fused_adam_utils.cuh
  114. fused_adamw_amsgrad_impl.cu
  115. fused_adamw_amsgrad_impl.cuh
  116. fused_adamw_impl.cu
  117. fused_adamw_impl.cuh
  118. FusedAdamKernel.cu
  119. FusedAdamWKernel.cu
  120. FusedSgdKernel.cu
  121. GcdLcmKernel.cu
  122. GridSampler.cpp
  123. GridSampler.cu
  124. GridSampler.cuh
  125. GridSampler.h
  126. group_norm_kernel.cu
  127. hermite_polynomial_h.cu
  128. hermite_polynomial_he.cu
  129. IGammaKernel.cu
  130. Im2Col.cu
  131. im2col.cuh
  132. Indexing.cu
  133. IndexKernel.cpp
  134. IndexKernel.cu
  135. IndexKernel.h
  136. int4mm.cu
  137. jit_utils.cpp
  138. jit_utils.h
  139. JitLoops.cuh
  140. KernelUtils.cuh
  141. laguerre_polynomial_l.cu
  142. LaunchUtils.h
  143. layer_norm_kernel.cu
  144. LegacyThrustHelpers.cu
  145. legendre_polynomial_p.cu
  146. Lerp.cu
  147. LinearAlgebra.cu
  148. LinearAlgebraStubs.cpp
  149. LogAddExpKernel.cu
  150. LogcumsumexpKernel.cu
  151. Loops.cuh
  152. Loss.cu
  153. LossCTC.cu
  154. Math.cuh
  155. MaxMinElementwiseKernel.cu
  156. MaxUnpooling.cu
  157. MemoryAccess.cuh
  158. MiscUtils.h
  159. MixedDtypesLinear.cu
  160. modified_bessel_i0.cu
  161. modified_bessel_i1.cu
  162. modified_bessel_k0.cu
  163. modified_bessel_k1.cu
  164. MultiLabelMarginCriterion.cu
  165. MultiMarginLoss.cu
  166. MultinomialKernel.cu
  167. MultiTensorApply.cuh
  168. NaiveConvolutionTranspose2d.cu
  169. NaiveConvolutionTranspose3d.cu
  170. NaiveDilatedConvolution.cu
  171. NLLLoss2d.cu
  172. Nonzero.cu
  173. Normalization.cu
  174. Normalization.cuh
  175. PersistentSoftmax.cuh
  176. PointwiseOpsKernel.cu
  177. Pow.cuh
  178. PowKernel.cu
  179. Randperm.cu
  180. Randperm.cuh
  181. RangeFactories.cu
  182. RecordStream.cu
  183. Reduce.cu
  184. Reduce.cuh
  185. ReduceAMinMaxKernel.cu
  186. ReduceArgMaxKernel.cu
  187. ReduceArgMinKernel.cu
  188. ReduceLogicKernel.cu
  189. ReduceMaxValuesKernel.cu
  190. ReduceMinValuesKernel.cu
  191. ReduceMomentKernel.cu
  192. ReduceNormKernel.cu
  193. ReduceOps.cpp
  194. ReduceOps.h
  195. ReduceSumProdKernel.cu
  196. reduction_template.cuh
  197. ReflectionPad.cu
  198. RenormKernel.cu
  199. Repeat.cu
  200. ReplicationPadding.cu
  201. Resize.cpp
  202. Resize.h
  203. RNN.cu
  204. RreluWithNoise.cu
  205. scaled_modified_bessel_k0.cu
  206. scaled_modified_bessel_k1.cu
  207. ScanKernels.cpp
  208. ScanKernels.h
  209. ScanUtils.cuh
  210. ScatterGatherKernel.cu
  211. SegmentReduce.cu
  212. Shape.cu
  213. shifted_chebyshev_polynomial_t.cu
  214. shifted_chebyshev_polynomial_u.cu
  215. shifted_chebyshev_polynomial_v.cu
  216. shifted_chebyshev_polynomial_w.cu
  217. SoftMax.cu
  218. Sort.cpp
  219. Sort.cu
  220. Sort.h
  221. SortImpl.cu
  222. Sorting.cpp
  223. Sorting.cu
  224. Sorting.h
  225. SortingCommon.cuh
  226. SortingRadixSelect.cuh
  227. SortStable.cu
  228. SortStable.h
  229. SortUtils.cuh
  230. SparseBinaryOpIntersectionKernel.cu
  231. SparseMM.cu
  232. SpectralOps.cpp
  233. SpectralOps.cu
  234. spherical_bessel_j0.cu
  235. StepKernel.cu
  236. SummaryOps.cu
  237. TensorCompare.cpp
  238. TensorCompare.cu
  239. TensorFactories.cu
  240. TensorModeKernel.cpp
  241. TensorModeKernel.cu
  242. TensorModeKernel.cuh
  243. TensorModeKernel.h
  244. TensorShape.cu
  245. TensorShapeCUDA.cpp
  246. TensorTopK.cpp
  247. TensorTopK.cu
  248. TensorTopK.h
  249. TensorTransformations.cu
  250. thread_constants.h
  251. TriangularOps.cu
  252. UnaryComplexKernels.cu
  253. UnaryFractionKernels.cu
  254. UnaryGammaKernels.cu
  255. UnaryGeometricAcoshKernel.cu
  256. UnaryGeometricAcosKernel.cu
  257. UnaryGeometricAsinhKernel.cu
  258. UnaryGeometricAsinKernel.cu
  259. UnaryGeometricAtanhKernel.cu
  260. UnaryGeometricAtanKernel.cu
  261. UnaryGeometricCoshKernel.cu
  262. UnaryGeometricCosKernel.cu
  263. UnaryGeometricSinhKernel.cu
  264. UnaryGeometricSinKernel.cu
  265. UnaryGeometricTanhKernel.cu
  266. UnaryGeometricTanKernel.cu
  267. UnaryLogKernels.cu
  268. UnaryOpsKernel.cu
  269. UnarySignKernels.cu
  270. UnarySpecialOpsKernel.cu
  271. UnfoldBackwardKernel.cu
  272. Unique.cu
  273. UniqueCub.cu
  274. UniqueCub.cuh
  275. UpSample.cuh
  276. UpSampleBicubic2d.cu
  277. UpSampleBilinear2d.cu
  278. UpSampleLinear1d.cu
  279. UpSampleNearest1d.cu
  280. UpSampleNearest2d.cu
  281. UpSampleNearest3d.cu
  282. UpSampleTrilinear3d.cu
  283. ValidateCompressedIndicesKernel.cu
  284. vol2col.cuh
  285. WeightNorm.cu
  286. ZetaKernel.cu