onnxruntime/onnxruntime/contrib_ops/cpu
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
..
attnlstm Introduce PrimitiveType into a Type System along with an integer constant (#2307) 2019-11-08 17:47:06 -08:00
bert Add SafeInt bounds checking to memory allocation size calculations. (#3022) 2020-02-20 11:41:03 -08:00
activations.cc fix Gelu CPU: remove MayInplace() declaration (#2306) 2019-11-01 18:10:05 -07:00
activations.h Adjust the grouping logic in ThreadPool::TryBatchParallelFor (#3207) 2020-03-13 12:49:17 -07:00
cdist.cc New OP: CDist (#1808) 2019-09-17 10:55:31 -07:00
cdist.h Move access to intra-op threadpool into OpKernelContext. (#2091) 2019-10-11 10:36:20 -07:00
conv_transpose_with_dynamic_pads.cc
conv_transpose_with_dynamic_pads.h
crop.cc
crop.h Replace GSL with GSL-LITE submodule and fix up refs (#1920) 2019-10-01 12:43:29 -07:00
crop_and_resize.cc Add BatchParallelFor, TryParallelFor, TryBatchParallelFor into ThreadPool (#2476) 2019-11-27 00:32:26 -08:00
crop_and_resize.h Replace GSL with GSL-LITE submodule and fix up refs (#1920) 2019-10-01 12:43:29 -07:00
dynamicslice.cc move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00
element_wise_ops.cc move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00
element_wise_ops.h move all contrib ops to contrib ops namespace (#1190) 2019-06-24 10:19:01 -07:00
expand_dims.cc
expand_dims.h
fused_activation.cc implement conv+clip fusion (#1412) 2019-07-17 12:16:45 -07:00
fused_activation.h cleanup fused conv activation handling (#1403) 2019-07-14 16:34:16 -07:00
fused_conv.cc cleanup fused conv activation handling (#1403) 2019-07-14 16:34:16 -07:00
fused_gemm.cc cleanup fused conv activation handling (#1403) 2019-07-14 16:34:16 -07:00
image_scaler.cc
image_scaler.h
layer_norm.cc Add SafeInt bounds checking to memory allocation size calculations. (#3022) 2020-02-20 11:41:03 -08:00
layer_norm.h Add layernorm operator (#1967) 2019-10-03 11:32:13 -07:00
matmul_integer16.cc Add support for int8 x uint8 for MatMulInteger, and int16 x int16 custom op (#1391) 2019-08-28 21:40:24 -07:00
matmul_integer16.h Add support for int8 x uint8 for MatMulInteger, and int16 x int16 custom op (#1391) 2019-08-28 21:40:24 -07:00
maxpool_with_mask.cc
maxpool_with_mask.h extract pooling attributes into a separate class (#1906) 2019-09-25 15:11:02 +10:00
mean_variance_normalization_exp.cc
murmur_hash3.cc Introduce container type runtime checks and other improvements (#2522) 2019-12-04 16:04:17 -08:00
murmur_hash3.h Improve code readability and performance. (#2257) 2019-10-25 16:19:59 -07:00
nchwc_ops.cc NCHWc Upsample/Mul optimizations (#3116) 2020-03-02 14:40:49 -08:00
nchwc_ops.h NCHWc Upsample/Mul optimizations (#3116) 2020-03-02 14:40:49 -08:00
quantize_ops.cc add dequantize and quantize back to contrib ops (#1712) 2019-09-06 08:55:42 -07:00
sample.cc
sample.h
skip_layer_norm.cc Implement AddGelu and SkipLayerNorm (#2487) 2019-11-28 08:29:59 -08:00
skip_layer_norm.h Implement AddGelu and SkipLayerNorm (#2487) 2019-11-28 08:29:59 -08:00
tokenizer.cc Introduce container type runtime checks and other improvements (#2522) 2019-12-04 16:04:17 -08:00
unique.cc
unique.h
word_conv_embedding.cc Move access to intra-op threadpool into OpKernelContext. (#2091) 2019-10-11 10:36:20 -07:00
word_conv_embedding.h Let mlas use session thread pool (#1609) 2019-08-16 13:21:15 -07:00