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