PR #3645 GatherND-12 Implementation

GatherND-12 Implementation #3645

pengwa merged 15 commits into ort_training from bahuang/fix_gathernd

Renamed, UT passing

e0a9dd31

Move GatherND CUDA Kerenl into onnxruntime

fb73cdec

Merge GatherNDOpTest

2894e686

Refactor Test code

8dd13afa

SherlockNoMad requested a review from

pengwa 5 years ago

SherlockNoMad requested a review 5 years ago

Merge CPU Kernel Impl

b13d9311

SherlockNoMad commented on 2020-04-23

Handle Negative Indice, Fix UT

c96f266f

Improve CUDA kernel to handle negative index

ee168ecd

SherlockNoMad requested a review from

souptc 5 years ago

Minor Fixes

3ed96fc7

SherlockNoMad commented on 2020-04-23

pengwa commented on 2020-04-23

pengwa dismissed these changes on 2020-04-23

pengwa commented on 2020-04-23

Preserve GatherND-1 Cuda kernel

61cd5aa3

SherlockNoMad dismissed their stale review via 61cd5aa3 5 years ago

Fix Mac build

13cc0b44

souptc commented on 2020-04-23

souptc dismissed these changes on 2020-04-23

pengwa dismissed these changes on 2020-04-24

fix UT

d3418ec8

SherlockNoMad dismissed their stale review via d3418ec8 5 years ago

Fix Build

881c418c

Merge branch 'ort_training' into bahuang/fix_gathernd

95c7c205

fix GatherNDOpTest.double > CUDA error cudaErrorInvalidDeviceFunction…

e45b09a8

Lafi7e approved these changes on 2020-04-24

Merge branch 'ort_training' of https://github.com/microsoft/onnxrunti…

d0e38227

pengwa merged b4d4ea2e into ort_training 5 years ago

pengwa deleted the bahuang/fix_gathernd branch 5 years ago

Reviewers

Lafi7e

pengwa

souptc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

onnxruntime GatherND-12 Implementation #3645 Merged

GatherND-12 Implementation #3645

onnxruntime
GatherND-12 Implementation
#3645

Merged