microsoft/onnxruntime

Pull Requests Commits

codemzs committed 5 years ago

d2734cff

codemzs committed 5 years ago

f7f14725

fix cuda dropout input for training_mode.

codemzs committed 5 years ago

6ea9f824

linux build break.

codemzs committed 5 years ago

06d33042

Merge branch 'ort_training' of https://github.com/microsoft/onnxruntime into function_body

codemzs committed 5 years ago

261b2f9c

Update dropout kernel with updated spec, add test infra for dropout tests.

codemzs committed 5 years ago

2138abf2

Support GPU Event Operators (#3653)

wschin committed 5 years ago

Verified 72b38f0a

Set gradient as output only for easy mode (#3694)

ashbhandare committed 5 years ago

Verified d06763ac

GatherND-12 Implementation (#3645)

SherlockNoMad committed 5 years ago

Verified b4d4ea2e

thrustallocator is not needed since cub is used directly for gather now. (#3683)

weixingzhang committed 5 years ago

Verified 2f8a17dc

type cast for ratio is not necessary for dropout (#3682)

weixingzhang committed 5 years ago

Verified c929963d

move cpu/cuda related files to coresponding cpu/cuda folder (#3668)

weixingzhang committed 5 years ago

Verified f4a04c04

Simplify and clean code (#3655)

weixingzhang committed 5 years ago

Verified 33662480

fixed mnist bug (#3569)

XiaocenDong committed 5 years ago

Verified 125f68f3

Add pipeline transformer for wait/record node (#3513)

xzhu1900 committed 5 years ago

Verified f1ba9aaf

GatherElementsGrad Kernels (#3627)

pengwa committed 5 years ago

Verified 6136fd07

Try not to modify base name (#3638)

wschin committed 5 years ago

Verified d9641f29

Expand elimination and Expand gradient. (#3610)

Lafi7e committed 5 years ago

Verified ffe19ae4

expose training session so the training app could register custom kernel and transformers (#3642)

souptc committed 5 years ago

Verified 37f4f743

fixes for ort_trainer.py to resume from checkpoint (#3510)

suffiank committed 5 years ago

Verified 0e12d05c

Refactoring code related to WARP_SIZE. (#3623)

weixingzhang committed 5 years ago

Verified e4fc8325

Merge pull request #3607 from microsoft/edgchen1/merge_from_master

edgchen1 committed 5 years ago

Verified bb9b0ba5

Add Lamb shape inference (#3634)

wschin committed 5 years ago

Verified ab70625b

Merge remote-tracking branch 'origin/master' into edgchen1/merge_from_master

edgchen1 committed 5 years ago

8df5076d

Merge remote-tracking branch 'origin/ort_training' into edgchen1/merge_from_master

edgchen1 committed 5 years ago

8d09cefa

Clean up OPTIONAL name conflict workarounds in ort_training. (#3622)

edgchen1 committed 5 years ago

Verified b518cb2a

Eliminate Useless Cast during Transformer. (#3606)

Lafi7e committed 5 years ago

Verified d3a2ac5c

Refine BERT optimization script options (#3618)

tianleiwu committed 5 years ago

Verified d69bc313

Merge branch 'ort_training' of https://github.com/microsoft/onnxruntime into function_body

codemzs committed 5 years ago

834c4267

Improve TopK performance. (#3612)

skottmckay committed 5 years ago

Verified b4508dbd

Older