onnxruntime
Softmax interface update
#12469
Merged

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