Google Git
Sign in
android / platform / external / pytorch / 92c9848c04
commit92c9848c04fe0b987487585dc5f0dde6ae269b5c[log] [tgz]
authorSam Gross <sgross@fb.com>Thu Oct 12 10:22:13 2017 -0700
committerZachary DeVito <zdevito@gmail.com>Thu Oct 12 19:45:15 2017 -0700
treeb1b7dae1934382a894cfb7a4c2d575dc4a9dab6e
parent5d689989ece5d7d04751e27f6c5ae8a7e927a397 [diff]
Support wrap_dim in nn.yaml
2 files changed
tree: b1b7dae1934382a894cfb7a4c2d575dc4a9dab6e
  1. templates/
  2. test/
  3. ArrayRef.h
  4. ATen.h
  5. ATenGeneral.h
  6. CheckGenerator.h
  7. CMakeLists.txt
  8. code_template.py
  9. common_with_cwrap.py
  10. Context.cpp
  11. Context.h
  12. copy_wrapper.py
  13. CPUFixedAllocator.h
  14. CPUGenerator.cpp
  15. CUDAFixedAllocator.h
  16. CUDAGenerator.cpp
  17. cwrap_parser.py
  18. Declarations.cwrap
  19. dispatch_macros.py
  20. DLConvertor.cpp
  21. DLConvertor.h
  22. dlpack.h
  23. ExpandUtils.h
  24. extract_cwrap.py
  25. Formatting.cpp
  26. Formatting.h
  27. function_wrapper.py
  28. gen.py
  29. Generator.h
  30. Half.h
  31. Local.cwrap
  32. nn.yaml
  33. nn_parse.py
  34. preprocess_declarations.py
  35. Scalar.cpp
  36. Scalar.h
  37. ScalarType.h
  38. SparseTensorRef.h
  39. Storage.h
  40. TensorAccessor.h
  41. TensorBase.h
  42. TensorImpl.h
  43. TensorOperators.h
  44. THLongStorageView.h
  45. Utils.cpp
  46. Utils.h
  47. WrapDimUtils.h