mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Summary: 1. Move ATen threadpool & open registration mechanism to C10 2. Move the `global_work_queue` to use this open registration mechanism, to allow users to substitute in their own Pull Request resolved: https://github.com/pytorch/pytorch/pull/17788 Reviewed By: zdevito Differential Revision: D14379707 Pulled By: jamesr66a fbshipit-source-id: 949662d0024875abf09907d97db927f160c54d45 |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| image_input_op_gpu.cc | ||
| image_input_op.cc | ||
| image_input_op.h | ||
| transform_gpu.cu | ||
| transform_gpu.h | ||