Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • 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
vision_cuda.h 3.11 KB