Skip to content

GitLab

Switch branch/tag
  1. 16 Oct, 2020 1 commit
    • bmanga's avatar
      Ensure torchvision operators are added in C++ (#2798) · adfc15c4
      bmanga authored 
      * Ensure torchvision operators are registered in C++ via weak symbols

* Add note to README on how to ensure that torchvision operators are available in C++

* Fix dllimport/dllexport on windows, format files

* Factor out common macros in single file

* Expose cuda_version in the API, use it to avoid pruning of ops initializer
      adfc15c4