ai.onnx.HardSwish

ai.onnx · standard ONNX operator · ONNX opset ≥ 14

Description

Applies the HardSwish activation elementwise: y = x * max(0, min(1, x/6 + 0.5)), which is equivalent to x * HardSigmoid(x) with alpha=1/6 and beta=0.5. The output has the same shape as the input.

See the ONNX HardSwish spec for the reference semantics.

Inputs

Name Bind key Logical dtype Rank Shape Description Presence
X x T Values transformed elementwise by the HardSwish activation. required

Outputs

Name Bind key Logical dtype Rank Shape Description Presence
Y y T same as X same as X Output tensor with HardSwish applied elementwise; 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/ai.onnx.HardSwish", { 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.