onnxruntime/onnxruntime/contrib_ops
Changming Sun 0a1257e467
Adjust the grouping logic in ThreadPool::TryBatchParallelFor (#3207)
1. No more plus 1.
2. Use MlasPartitionWork function to calculate the work index range.
2020-03-13 12:49:17 -07:00
..
cpu Adjust the grouping logic in ThreadPool::TryBatchParallelFor (#3207) 2020-03-13 12:49:17 -07:00
cuda Add support for loading TensorProtos with external data from optimizer Initializer (#3045) 2020-02-28 13:19:16 -08: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