tree: 7743ecad5232e0e762e91310d98b16c20d63bbf6 [path history] [tgz]
  1. benchmarks/
  2. core/
  3. cpu/
  4. cuda/
  5. cudnn/
  6. detail/
  7. functorch/
  8. hip/
  9. metal/
  10. miopen/
  11. mkl/
  12. mps/
  13. native/
  14. nnapi/
  15. ops/
  16. quantized/
  17. templates/
  18. test/
  19. vulkan/
  20. xpu/
  21. .gitignore
  22. AccumulateType.cpp
  23. AccumulateType.h
  24. ArrayRef.h
  25. ATen.h
  26. ATenConfig.cmake.in
  27. autocast_mode.cpp
  28. autocast_mode.h
  29. Backend.h
  30. Backtrace.h
  31. BlasBackend.h
  32. CachedTensorUtils.cpp
  33. CachedTensorUtils.h
  34. ceil_div.h
  35. CMakeLists.txt
  36. code_template.h
  37. CollapseDims.h
  38. Config.h.in
  39. ConjugateFallback.cpp
  40. Context.cpp
  41. Context.h
  42. cpp_custom_type_hack.h
  43. CPUApplyUtils.h
  44. CPUFixedAllocator.h
  45. CPUGeneratorImpl.cpp
  46. CPUGeneratorImpl.h
  47. Device.h
  48. DeviceAccelerator.cpp
  49. DeviceAccelerator.h
  50. DeviceGuard.h
  51. Dimname.h
  52. DimVector.h
  53. Dispatch.cpp
  54. Dispatch.h
  55. Dispatch_v2.h
  56. div_rtn.h
  57. DLConvertor.cpp
  58. DLConvertor.h
  59. dlpack.h
  60. DynamicLibrary.cpp
  61. DynamicLibrary.h
  62. EmptyTensor.cpp
  63. EmptyTensor.h
  64. ExpandBase.h
  65. ExpandUtils.cpp
  66. ExpandUtils.h
  67. Formatting.h
  68. function_wrapper.py
  69. FunctionalInverses.cpp
  70. FunctionalizeFallbackKernel.cpp
  71. FunctionalStorageImpl.cpp
  72. FunctionalStorageImpl.h
  73. FunctionalTensorWrapper.cpp
  74. FunctionalTensorWrapper.h
  75. FuncTorchTLS.cpp
  76. FuncTorchTLS.h
  77. Generator.h
  78. InferSize.h
  79. InitialTensorOptions.h
  80. jit_macros.h
  81. jiterator_macros.h
  82. Layout.h
  83. LegacyBatchedFallback.cpp
  84. LegacyBatchedFallback.h
  85. LegacyBatchedTensorImpl.cpp
  86. LegacyBatchedTensorImpl.h
  87. LegacyBatchingRegistrations.cpp
  88. LegacyVmapMode.cpp
  89. LegacyVmapMode.h
  90. LegacyVmapTransforms.cpp
  91. LegacyVmapTransforms.h
  92. LinalgBackend.h
  93. MapAllocator.cpp
  94. MapAllocator.h
  95. MatrixRef.h
  96. MemoryOverlap.cpp
  97. MemoryOverlap.h
  98. NamedTensor.h
  99. NamedTensorUtils.cpp
  100. NamedTensorUtils.h
  101. NestedTensorImpl.cpp
  102. NestedTensorImpl.h
  103. NumericUtils.h
  104. OpaqueTensorImpl.h
  105. OpMathType.h
  106. PadNd.h
  107. Parallel-inl.h
  108. Parallel.h
  109. ParallelCommon.cpp
  110. ParallelFuture.h
  111. ParallelNative.cpp
  112. ParallelNative.h
  113. ParallelOpenMP.cpp
  114. ParallelOpenMP.h
  115. ParallelThreadPoolNative.cpp
  116. PTThreadPool.h
  117. PythonTorchFunctionTLS.cpp
  118. PythonTorchFunctionTLS.h
  119. record_function.cpp
  120. record_function.h
  121. SavedTensorHooks.cpp
  122. SavedTensorHooks.h
  123. Scalar.h
  124. ScalarOps.cpp
  125. ScalarOps.h
  126. ScalarType.h
  127. SequenceNumber.cpp
  128. SequenceNumber.h
  129. SmallVector.h
  130. SparseCsrTensorImpl.cpp
  131. SparseCsrTensorImpl.h
  132. SparseCsrTensorUtils.h
  133. SparseTensorImpl.cpp
  134. SparseTensorImpl.h
  135. Storage.h
  136. StorageUtils.cpp
  137. StorageUtils.h
  138. Tensor.h
  139. TensorAccessor.h
  140. TensorGeometry.cpp
  141. TensorGeometry.h
  142. TensorIndexing.cpp
  143. TensorIndexing.h
  144. TensorIterator.cpp
  145. TensorIterator.h
  146. TensorIteratorInternal.h
  147. TensorMeta.cpp
  148. TensorMeta.h
  149. TensorNames.cpp
  150. TensorNames.h
  151. TensorOperators.h
  152. TensorOptions.h
  153. TensorSubclassLikeUtils.h
  154. TensorUtils.cpp
  155. TensorUtils.h
  156. ThreadLocalPythonObjects.cpp
  157. ThreadLocalPythonObjects.h
  158. ThreadLocalState.cpp
  159. ThreadLocalState.h
  160. TracerMode.h
  161. TypeDefault.h
  162. Utils.cpp
  163. Utils.h
  164. Version.cpp
  165. Version.h
  166. VmapModeRegistrations.cpp
  167. WrapDimUtils.h
  168. WrapDimUtilsMulti.h
  169. ZeroTensorFallback.cpp