onnxruntime/onnxruntime/contrib_ops
Changming Sun 06fc9506fd
Thread pool changes (#3153)
1. Copy tensorflow's thread pool class to ORT, so that we can get a better implementation of thread pool based parallelfor
2. Copy Eigen's thread pool class to ORT
3. Support thread affinity
4. Remove RNN kernel’s private thread pool
5. Modify pool kernels to use the thread pool when openmp is disabled.
2020-03-30 12:18:40 -07:00
..
cpu Thread pool changes (#3153) 2020-03-30 12:18:40 -07:00
cuda Replacing CudaAsyncBuffer with TArray to improve perf (#3303) 2020-03-24 12:13:27 -07:00
cpu_contrib_kernels.cc NCHWc Upsample/Mul optimizations (#3116) 2020-03-02 14:40:49 -08:00
cpu_contrib_kernels.h Fix kernel registry bug (#2137) 2019-10-17 23:10:54 -07:00
cuda_contrib_kernels.cc Revert an op version change (#3026) 2020-02-18 09:43:18 -08:00
cuda_contrib_kernels.h move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00