saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[js/webgpu] fix pad operator (#17775)

Browse source 
fix pad operator
This commit is contained in:
Guenther Schmuelling 2023-10-03 13:39:50 -07:00 committed by GitHub
parent d0519a7603
commit f8a8452a6b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
js/web/lib/wasm/jsep/webgpu/ops/pad.ts
View file

@ -209,7 +209,7 @@ const createPadProgramInfo =
const createPadAttributesFromInputs = (inputs: readonly TensorView[], attributes: PadAttributes): PadAttributes => {
if (inputs.length > 1) {
const bigInt64Pads = inputs[1].getBigInt64Array();
const value = (inputs.length >= 3) ? inputs[2].getFloat32Array()[0] : 0.0;
const value = (inputs.length >= 3 && inputs[2].data) ? inputs[2].getFloat32Array()[0] : 0.0;
const inputRank = inputs[0].dims.length;
const updatePads = new Int32Array(2 * inputRank).fill(0);
@ -220,7 +220,7 @@ const createPadAttributesFromInputs = (inputs: readonly TensorView[], attributes
updatePads[Number(axes[i]) + inputRank] = Number(bigInt64Pads[i + axes.length]);
}
} else {
bigInt64Pads.forEach((i, v) => updatePads[Number(i)] = (Number(v)));
bigInt64Pads.forEach((v, i) => updatePads[Number(i)] = (Number(v)));
}
const pads: number[] = [];