We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d31c32d commit f497e78Copy full SHA for f497e78
1 file changed
onnxruntime/core/providers/webgpu/math/softmax.cc
@@ -57,6 +57,7 @@ static std::string MaxVector(const std::string& name, int components) {
57
}
58
59
60
+// Online Softmax implementation as described in https://arxiv.org/abs/1805.02867
61
Status SoftmaxProgram::GenerateShaderCode(ShaderHelper& shader) const {
62
// Add input and output variables
63
const auto& input = shader.AddInput("x", ShaderUsage::UseUniform | ShaderUsage::UseIndicesTypeAlias |
0 commit comments