xla
b8401372 - lower NativeDropoutBackward (#5642) and lower native_dropout (#5643) (#5647)

Commit
2 years ago
lower NativeDropoutBackward (#5642) and lower native_dropout (#5643) (#5647) * lower NativeDropoutBackward (#5642) * lower NativeDropoutBackward * fix lowering and add python test * lower native_dropout (#5643) * prototype version (compiling error) * Add native_dropout manual lowering. * fix to tensor IR and add a simple native_dropout test * fix data type issue and update test case * fix IR hash issue * fix corner case when probability==0 * remove typo line * add test case when probability=0 --------- Co-authored-by: JackCaoG <59073027+JackCaoG@users.noreply.github.com> Co-authored-by: zpcore <piz@google.com>
Author
Parents
Loading