tree: 75138834d108011a815d2c67e7490ffb663ff978 [path history] [tgz]
  1. benchmarks/
  2. core/
  3. cpu/
  4. cuda/
  5. cudnn/
  6. detail/
  7. hip/
  8. metal/
  9. miopen/
  10. mkl/
  11. native/
  12. quantized/
  13. templates/
  14. test/
  15. vulkan/
  16. .gitignore
  17. AccumulateType.h
  18. ArrayRef.h
  19. ATen.h
  20. ATenConfig.cmake.in
  21. autocast_mode.cpp
  22. autocast_mode.h
  23. Backend.h
  24. Backtrace.h
  25. BatchedFallback.cpp
  26. BatchedFallback.h
  27. BatchedTensorImpl.cpp
  28. BatchedTensorImpl.h
  29. BatchingRegistrations.cpp
  30. CMakeLists.txt
  31. Config.h.in
  32. Context.cpp
  33. Context.h
  34. cpp_custom_type_hack.h
  35. CPUApplyUtils.h
  36. CPUFixedAllocator.h
  37. CPUGeneratorImpl.cpp
  38. CPUGeneratorImpl.h
  39. CUDAGeneratorImpl.h
  40. Device.h
  41. DeviceGuard.h
  42. Dimname.h
  43. DimVector.h
  44. Dispatch.h
  45. div_rtn.h
  46. DLConvertor.cpp
  47. DLConvertor.h
  48. dlpack.h
  49. DynamicLibrary.cpp
  50. DynamicLibrary.h
  51. ExpandUtils.cpp
  52. ExpandUtils.h
  53. Formatting.h
  54. function_wrapper.py
  55. gen_vulkan_glsl.py
  56. gen_vulkan_spv.py
  57. Generator.h
  58. InferSize.h
  59. InitialTensorOptions.h
  60. Layout.h
  61. LegacyTHFunctionsCPU.cpp
  62. LegacyTHFunctionsCPU.h
  63. LegacyTHFunctionsCUDA.h
  64. MatrixRef.h
  65. MemoryOverlap.cpp
  66. MemoryOverlap.h
  67. NamedTensor.h
  68. NamedTensorUtils.cpp
  69. NamedTensorUtils.h
  70. NumericUtils.h
  71. OpaqueTensorImpl.h
  72. Parallel.h
  73. ParallelCommon.cpp
  74. ParallelNative.cpp
  75. ParallelNative.h
  76. ParallelNativeTBB.cpp
  77. ParallelNativeTBB.h
  78. ParallelOpenMP.cpp
  79. ParallelOpenMP.h
  80. ParallelThreadPoolNative.cpp
  81. PTThreadPool.h
  82. record_function.cpp
  83. record_function.h
  84. Scalar.h
  85. ScalarOps.cpp
  86. ScalarOps.h
  87. ScalarType.h
  88. SequenceNumber.cpp
  89. SequenceNumber.h
  90. SmallVector.h
  91. SparseTensorImpl.cpp
  92. SparseTensorImpl.h
  93. SparseTensorUtils.h
  94. Storage.h
  95. Tensor.h
  96. TensorAccessor.h
  97. TensorGeometry.cpp
  98. TensorGeometry.h
  99. TensorIndexing.cpp
  100. TensorIndexing.h
  101. TensorNames.cpp
  102. TensorNames.h
  103. TensorOperators.h
  104. TensorOptions.h
  105. TensorUtils.cpp
  106. TensorUtils.h
  107. ThreadLocalState.cpp
  108. ThreadLocalState.h
  109. TracerMode.h
  110. TypeDefault.h
  111. Utils.cpp
  112. Utils.h
  113. Version.cpp
  114. Version.h
  115. VmapMode.cpp
  116. VmapMode.h
  117. VmapModeRegistrations.cpp
  118. VmapTransforms.cpp
  119. VmapTransforms.h
  120. WrapDimUtils.h
  121. WrapDimUtilsMulti.h