saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-02 03:55:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add missing GELU to ApiHelpers.h

Browse source
This commit is contained in:
Dwayne Robinson 2022-06-14 23:28:15 -07:00
parent 4c1a410d54
commit e3ec30efb6
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/dml/DmlExecutionProvider/src/External/DirectMLHelpers/ApiHelpers.h vendored
View file

@ -28,6 +28,7 @@ union ActivationOperatorDescUnion
DML_ACTIVATION_TANH_OPERATOR_DESC tanh;
DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC thresholdedRelu;
DML_ACTIVATION_SHRINK_OPERATOR_DESC shrink;
DML_ACTIVATION_GELU_OPERATOR_DESC gelu;
};
struct ActivationOperatorDesc
@ -64,6 +65,7 @@ struct ActivationOperatorDesc
case DML_OPERATOR_ACTIVATION_TANH: return { activationType, &params.tanh };
case DML_OPERATOR_ACTIVATION_THRESHOLDED_RELU: return { activationType, &params.thresholdedRelu };
case DML_OPERATOR_ACTIVATION_SHRINK: return { activationType, &params.shrink };
case DML_OPERATOR_ACTIVATION_GELU: return { activationType, &params.gelu };
default:
ORT_THROW_HR(E_INVALIDARG);
return { activationType, &params.relu };