com.microsoft.FastGelu

com.microsoft · ONNX Runtime contrib operator · contrib since_version 1

Description

Applies the GELU (Gaussian Error Linear Unit) activation using a tanh approximation: Y = 0.5 * X * (1 + tanh(0.797885 * X + 0.035677 * X^3)). An optional bias is added to X before the activation is computed. This WebGPU package implements float16 and float32; the schema-allowed double and bfloat16 types are not supported.

See the ONNX Runtime FastGelu contrib-operator spec for the reference semantics.

Inputs

Name Bind key Logical dtype Rank Shape Description Presence
X X T Values transformed by FastGelu after adding the optional bias. required
bias bias T 1 Optional 1-D bias added to X along the last dimension before the GELU activation. optional

Outputs

Name Bind key Logical dtype Rank Shape Description Presence
Y Y T same as X same as X Output tensor after applying the GELU activation; same shape as X. required

Type constraints

Variable Allowed dtypes
T float32, float16

Files

Use with @huggingface/kernels

The loader derives every required output's shape and logical dtype from the manifest contract and this call. It then allocates the result tensors automatically.

The version: 1 option selects the published kernel contract; it is independent of any operator opset, contrib since_version, or model version.

Replace each *Data placeholder with a typed array containing the corresponding input data.

import { getKernel } from "@huggingface/kernels";

const kernel = await getKernel("webgpu-kernels/com.microsoft.FastGelu", { version: 1 });
const { Y } = await kernel({ X: { data: XData, shape: [5] } });
Downloads last month
-
kernel
webgpu
wgsl
apache-2.0
WebGPU

Requires WebGPU support. See the compatibility table.