onnxruntime
Softmax interface update
#12469
Merged
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
5
Changes
View On
GitHub
Commits
Template datatype for SoftmaxWithRawMaskSmallKernel in ROCm EP
groenenboomj
committed
4 years ago
Remove valid_items usage from SoftmaxWithRawMaskSmallKernel for ROCm EP
groenenboomj
committed
3 years ago
Update accumulator type in ROCm EP for SoftmaxWithRawMaskSmallKernel
groenenboomj
committed
3 years ago
Reset casting to old behavior
groenenboomj
committed
3 years ago
Document steps to optimize SoftMax kernel on ROCm EP
groenenboomj
committed
3 years ago
Loading