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

fix ConvTranspose 1D (#20194)

Browse source
This commit is contained in:
Guenther Schmuelling 2024-04-05 10:05:32 -07:00 committed by GitHub
parent 4f2d454211
commit c529e05e38
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
js/web/lib/wasm/jsep/webgpu/ops/conv-transpose.ts
View file

@ -268,7 +268,7 @@ const convTranspose1d = (context: ComputeContext, attributes: ConvTransposeAttri
//[FILTER_OUT_CHANNEL, FILTER_IN_CHANNEL, kW] -> [FILTER_OUT_CHANNEL, FILTER_IN_CHANNEL, kH=1, kW]
context.inputs[1].reshape([context.inputs[1].dims[0], context.inputs[1].dims[1], 1, context.inputs[1].dims[2]])
];
if (inputs.length === 3) {
if (context.inputs.length === 3) {
inputs.push(context.inputs[2]);
}
let kernelShape = attributes.kernelShape;

2
onnxruntime/core/providers/js/operators/conv_transpose.h
View file

@ -44,7 +44,7 @@ class ConvTranspose : public JsKernel {
"autoPad" : $1,
"dilations" : [$2],
"group" : $3,
"kernel_shape" : [$4],
"kernelShape" : [$4],
"pads" : [ $5, $6 ],
"strides" : [$7],
"wIsConst" : () JS_ARROW(!!HEAP8[$9]),