saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[JS/Web] Bug fix for Reshape Pack (#7754)

Browse source
This commit is contained in:
Du Li 2021-05-19 11:15:47 -07:00 committed by GitHub
parent 203efef147
commit fbe6eccc65
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
js/web/lib/onnxjs/backends/webgl/ops/reshape-packed.ts
View file

@ -30,6 +30,11 @@ export class WebGLReshapePacked extends Reshape implements WebGLOperator {
// TODO: optimize the algorithm -- in some cases, if the last two dims are
// the same between input shape and output shape, the packed reshape can be
// treated as no-op.
// TODO: the implementation is a bit complicated due to the fact tensor shape is
// immutable once the tensor is created, plus the tensor shape has a 1-to-1
// mapping with texture layout. In the future, we may consider relaxing this
// assumption.
const originInputShape = inputs[0].dims;
this.inputShape3D = processDims3D(inputs[0].dims);
let inputLayout: TextureLayout;
@ -138,6 +143,9 @@ export class WebGLReshapePacked extends Reshape implements WebGLOperator {
inputTDs = [originalInputTD];
}
let outputLayout = this.outputLayout;
// Use original output shape for runData output layout.
outputLayout.shape = this.outputShape;
outputLayout.unpackedShape = this.outputShape;
if (outputLayout === undefined) {
outputLayout = handler.createTextureLayoutFromShape(
this.outputShape, 4, this.outputShape, {isPacked: true, reverseWH: true});