onnxruntime/onnxruntime/test
Vincent Wang 0fa3aeb65c
[CUDA] Add Strided Tensor Support for Expand->GatherElements for Training (#11976)
* strided tensor for expand and gather_elements

* bugfix

* simplify CoalesceDimensions

* resolve comments

* resolve more comments.
2022-07-25 16:05:26 +08:00
..
api_tests_without_env Remove OpenMP code (#10032) 2021-12-15 00:58:42 -08:00
common [CUDA] Add Strided Tensor Support for Expand->GatherElements for Training (#11976) 2022-07-25 16:05:26 +08:00
contrib_ops x86/64 U8S8 Gemm Precision Fix (#12088) 2022-07-13 10:12:25 -07:00
debug_node_inputs_outputs Enable more static analysis warnings and enable the analyzer for training cpu (#10176) 2022-01-27 11:17:20 -08:00
eager Revert "fixed point based requantization on arm64 (#11540)" (#11732) 2022-06-03 19:12:25 -07:00
framework Restructure function inliner (#11731) 2022-06-24 09:21:31 -07:00
fuzzing
global_thread_pools Remove OpenMP code (#10032) 2021-12-15 00:58:42 -08:00
ir Add better error message for subgraph output coming directly from outer scope value. (#11638) 2022-05-28 08:04:27 +10:00
mlas fuse Conv+Add+activation for CPU from different op-branch (#10987) 2022-04-01 09:25:17 +08:00
onnx Deprecate APIs returning raw ptrs and provide replacements (#11922) 2022-06-24 09:50:04 -07:00
opaque_api Deprecate APIs returning raw ptrs and provide replacements (#11922) 2022-06-24 09:50:04 -07:00
optimizer x86/64 U8S8 Gemm Precision Fix (#12088) 2022-07-13 10:12:25 -07:00
perftest Initialize generated tensor data in onnxruntime_perf_test. (#12275) 2022-07-22 16:26:13 -07:00
platform Add file mapping for windows platform. (#12183) 2022-07-18 09:24:12 -07:00
proto
providers [CUDA] Add Strided Tensor Support for Expand->GatherElements for Training (#11976) 2022-07-25 16:05:26 +08:00
python support gpt2 model with greedy search (#12068) 2022-07-22 15:45:16 -07:00
quantization
shared_lib Fix Reduced Ops build pipeline (#12144) 2022-07-11 19:02:38 -07:00
testdata CPU AdamW implementation (#11978) 2022-07-25 09:43:52 +08:00
unittest_main
util EP factory creation cleanup and enhancements. (#11798) 2022-06-16 07:01:41 +10:00
wasm Add a build option to create a WebAssembly static library (#10184) 2022-01-18 18:05:04 -08:00
win_getopt
xctest