Hugging Face's logo

Kernels:
webgpu-kernels
/
ai.onnx.GatherElements

Kernel card Files
xet
 Community
Xenova HF Staff commited on
Commit
828fdc4
·
verified ·
1 Parent(s): f00338a

sync c928d21e6cc1

Browse files
Files changed (2) hide show
  1. build/webgpu/bench.json +3 -3
  2. build/webgpu/metadata.json +3 -3
build/webgpu/bench.json CHANGED
@@ -7,7 +7,7 @@
7
  "attrs": { "axis": 1 },
8
  "inputs": {
9
  "data": { "dtype": "float32", "shape": [1024, 768] },
10
- "indices": { "dtype": "int32", "shape": [1024, 128] }
11
  },
12
  "outputs": { "output": { "dtype": "float32", "shape": [1024, 128] } },
13
  "bench": {
@@ -20,7 +20,7 @@
20
  "vars": { "dtype": "float32", "count": 17301504 },
21
  "inputs": {
22
  "data": { "dtype": "float32", "shape": [16, 524288] },
23
- "indices": { "dtype": "int32", "shape": [33, 524288] }
24
  },
25
  "outputs": { "output": { "dtype": "float32", "shape": [33, 524288] } },
26
  "bench": {
@@ -34,7 +34,7 @@
34
  "vars": { "dtype": "float16", "count": 131072 },
35
  "inputs": {
36
  "data": { "dtype": "float16", "shape": [1024, 768] },
37
- "indices": { "dtype": "int32", "shape": [1024, 128] }
38
  },
39
  "outputs": { "output": { "dtype": "float16", "shape": [1024, 128] } },
40
  "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * (dtypeBytes(args.dtype) * 2 + 4)" }] }
 
7
  "attrs": { "axis": 1 },
8
  "inputs": {
9
  "data": { "dtype": "float32", "shape": [1024, 768] },
10
+ "indices": { "dtype": "int32", "shape": [1024, 128], "min": 0, "max": 768 }
11
  },
12
  "outputs": { "output": { "dtype": "float32", "shape": [1024, 128] } },
13
  "bench": {
 
20
  "vars": { "dtype": "float32", "count": 17301504 },
21
  "inputs": {
22
  "data": { "dtype": "float32", "shape": [16, 524288] },
23
+ "indices": { "dtype": "int32", "shape": [33, 524288], "min": 0, "max": 16 }
24
  },
25
  "outputs": { "output": { "dtype": "float32", "shape": [33, 524288] } },
26
  "bench": {
 
34
  "vars": { "dtype": "float16", "count": 131072 },
35
  "inputs": {
36
  "data": { "dtype": "float16", "shape": [1024, 768] },
37
+ "indices": { "dtype": "int32", "shape": [1024, 128], "min": 0, "max": 768 }
38
  },
39
  "outputs": { "output": { "dtype": "float16", "shape": [1024, 128] } },
40
  "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * (dtypeBytes(args.dtype) * 2 + 4)" }] }
build/webgpu/metadata.json CHANGED
@@ -1,19 +1,19 @@
1
  {
2
  "name": "ai.onnx.GatherElements",
3
- "id": "_ai_onnx_gatherelements_webgpu_e9860f1",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "backend": { "type": "webgpu" },
7
  "digest": {
8
  "algorithm": "sha256",
9
  "files": {
10
- "bench.json": "bcWRb9m2vxl/bwvxTpx/zK9W0YkNoXzNdIpqOJOe2CI=",
11
  "gather-elements-rank2-vec4.wgsl.jinja": "0giK9mU05ulbta7kCji4+1HiXa2bhyzRKxw55Mu1GSA=",
12
  "gather-elements.wgsl.jinja": "otXACNTWXobPOOibQfwABUFgIoTL10tSuQNU6kAs5hQ=",
13
  "manifest.json": "FgK/3qWO+NN3odHqs1vavhKdhB7FwvlTQYpWhxvj2Hk=",
14
  "test.json": "s4kSX9LtX1rfFfk+Aiw+Y9YPuEgts9zcG/v94b/oJDs="
15
  }
16
  },
17
- "provenance": { "kernel": { "sha": "2e7068faf55e7f43df740015f6d1ee49391a41c5", "dirty": false } },
18
  "webgpu": { "manifestSpec": "1.0", "specialized": true, "opPath": "ops/ai.onnx.GatherElements" }
19
  }
 
1
  {
2
  "name": "ai.onnx.GatherElements",
3
+ "id": "_ai_onnx_gatherelements_webgpu_d82e03b",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "backend": { "type": "webgpu" },
7
  "digest": {
8
  "algorithm": "sha256",
9
  "files": {
10
+ "bench.json": "+6XI6fYMzJWO0lYAcf9ru+ZWTyUQoAlgfX1yhuQUOj4=",
11
  "gather-elements-rank2-vec4.wgsl.jinja": "0giK9mU05ulbta7kCji4+1HiXa2bhyzRKxw55Mu1GSA=",
12
  "gather-elements.wgsl.jinja": "otXACNTWXobPOOibQfwABUFgIoTL10tSuQNU6kAs5hQ=",
13
  "manifest.json": "FgK/3qWO+NN3odHqs1vavhKdhB7FwvlTQYpWhxvj2Hk=",
14
  "test.json": "s4kSX9LtX1rfFfk+Aiw+Y9YPuEgts9zcG/v94b/oJDs="
15
  }
16
  },
17
+ "provenance": { "kernel": { "sha": "c928d21e6cc1310861cba3bafb75f5f679ecf5f3", "dirty": false } },
18
  "webgpu": { "manifestSpec": "1.0", "specialized": true, "opPath": "ops/ai.onnx.GatherElements" }
19
  }