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