onnxruntime/onnxruntime/test
Bowen Bao 51c12a715b
Add NGramRepeatBlock contrib op (#8078)
**Description**: 
Enforce no repetition of n-grams. Scores are set to `-inf` for tokens that form a repeated n-gram if added to the back of the input_ids.

**Motivation and Context**
Needed by transformer models in sequence generation algorithms (greedy search and beam search). This module has heavy impact on performance, and can be highly parallelized.
2021-06-21 10:21:48 -07:00
..
api_tests_without_env
common Create and move quantization tests to a shared Quantized utils file. (#8054) 2021-06-16 17:00:36 -05:00
contrib_ops Add NGramRepeatBlock contrib op (#8078) 2021-06-21 10:21:48 -07:00
debug_node_inputs_outputs
eager support pass in custom op registry for eager mode (#8087) 2021-06-20 13:38:09 -07:00
featurizers_ops
framework Make logic in InsertCastTransformer around forcing a node to fp32 more precise. (#8018) 2021-06-11 13:54:40 +10:00
fuzzing
global_thread_pools
ir
mlas
onnx
opaque_api
optimizer Eliminate no op node - add 0 (#7798) 2021-05-25 13:01:34 -07:00
perftest
platform Add podspec template for ios package, update build settings (#7907) 2021-06-02 11:30:05 -07:00
proto
providers [NNAPI EP] Enable Slice support (#8031) 2021-06-17 12:36:12 -07:00
python Fix unit test typo in test_op_embed_layernorm.py (#8056) 2021-06-15 15:27:44 -05:00
shared_lib [js/react_native] Use a mobile ORT instead of a full ORT (#8042) 2021-06-15 13:36:05 -07:00
testdata Add NGramRepeatBlock contrib op (#8078) 2021-06-21 10:21:48 -07:00
tvm
unittest_main
util Re-enable some of the recently disabled cuda tests (#7873) 2021-06-03 14:28:30 -07:00
wasm [wasm] fix test report generation (#7953) 2021-06-04 18:05:17 -07:00
win_getopt
xctest