Skip to content

Commit f497e78

Browse files
author
Prathik Rao
committed
nit
1 parent d31c32d commit f497e78

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

onnxruntime/core/providers/webgpu/math/softmax.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ static std::string MaxVector(const std::string& name, int components) {
5757
}
5858
}
5959

60+
// Online Softmax implementation as described in https://arxiv.org/abs/1805.02867
6061
Status SoftmaxProgram::GenerateShaderCode(ShaderHelper& shader) const {
6162
// Add input and output variables
6263
const auto& input = shader.AddInput("x", ShaderUsage::UseUniform | ShaderUsage::UseIndicesTypeAlias |

0 commit comments

Comments
 (0)