| .. |
|
attnlstm
|
Thread pool changes (#3153)
|
2020-03-30 12:18:40 -07:00 |
|
bert
|
Remove eigen device from thread pool
|
2020-05-01 02:21:57 -07:00 |
|
activations.cc
|
Remove FastGelu from activations.
|
2020-04-09 19:17:54 -07:00 |
|
activations.h
|
Add omp impl for tryparallelfor and modify gelu to use fastgelu impl. (#3667)
|
2020-04-23 18:24:46 -07:00 |
|
cdist.cc
|
Rework CDist (#3393)
|
2020-04-09 14:05:25 +10:00 |
|
cdist.h
|
Rework CDist (#3393)
|
2020-04-09 14:05:25 +10:00 |
|
conv_transpose_with_dynamic_pads.cc
|
|
|
|
conv_transpose_with_dynamic_pads.h
|
|
|
|
cpu_contrib_kernels.cc
|
move cpu/cuda related files to coresponding cpu/cuda folder (#3668)
|
2020-04-24 00:12:02 -07:00 |
|
cpu_contrib_kernels.h
|
move cpu/cuda related files to coresponding cpu/cuda folder (#3668)
|
2020-04-24 00:12:02 -07:00 |
|
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
|
Thread pool changes (#3153)
|
2020-03-30 12:18:40 -07: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
|
Fix dynamicslice.cc after merge.
|
2020-04-09 15:17:21 -07:00 |
|
element_wise_ops.cc
|
|
|
|
element_wise_ops.h
|
|
|
|
expand_dims.cc
|
|
|
|
expand_dims.h
|
|
|
|
fused_activation.cc
|
|
|
|
fused_activation.h
|
|
|
|
fused_conv.cc
|
|
|
|
fused_gemm.cc
|
|
|
|
image_scaler.cc
|
|
|
|
image_scaler.h
|
|
|
|
inverse.cc
|
Implement Inverse(12) for CPU and CUDA (#3485)
|
2020-04-18 17:10:21 -07:00 |
|
layer_norm.cc
|
Thread pool changes (#3153)
|
2020-03-30 12:18:40 -07:00 |
|
layer_norm.h
|
Add layernorm operator (#1967)
|
2019-10-03 11:32:13 -07:00 |
|
matmul_integer16.cc
|
|
|
|
matmul_integer16.h
|
|
|
|
maxpool_with_mask.cc
|
|
|
|
maxpool_with_mask.h
|
Thread pool changes (#3153)
|
2020-03-30 12:18:40 -07: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
|
Make De/QuantizeLinear support half (#3531)
|
2020-04-17 12:17:48 -07:00 |
|
sample.cc
|
|
|
|
sample.h
|
|
|
|
skip_layer_norm.cc
|
Thread pool changes (#3153)
|
2020-03-30 12:18:40 -07: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 |
|
transpose_matmul.cc
|
address PR comments (#3312)
|
2020-03-25 19:35:12 -07:00 |
|
transpose_matmul.h
|
address PR comments (#3312)
|
2020-03-25 19:35:12 -07: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
|
|
|