onnxruntime/tools/ci_build
Sherlock 4bc17ca04e
CUDA ConvGrad Kernel (#7227)
* ConvGrad CUDA impl

* Set up the test case for Deberta Conv1D

* Add fp16 test

Co-authored-by: Sherlock Huang <bahuang@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
2021-04-06 22:09:06 -07:00
..
github build ONNXRuntime into WebAssembly (#6478) 2021-04-06 16:18:10 -07:00
wasm build ONNXRuntime into WebAssembly (#6478) 2021-04-06 16:18:10 -07:00
__init__.py
amd_hipify.py CUDA ConvGrad Kernel (#7227) 2021-04-06 22:09:06 -07:00
build.py build ONNXRuntime into WebAssembly (#6478) 2021-04-06 16:18:10 -07:00
clean_docker_image_cache.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
coverage.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
gen_def.py
get_docker_image.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
logger.py
op_registration_utils.py Support required types when excluding typed registrations (#6871) 2021-03-08 08:22:07 -08:00
op_registration_validator.py Support specifying globally allowed types from build script (#6677) 2021-02-22 14:05:00 -08:00
reduce_op_kernels.py Support required types when excluding typed registrations (#6871) 2021-03-08 08:22:07 -08:00