onnxruntime

1
// Copyright (c) Microsoft Corporation. All rights reserved.
2
// Licensed under the MIT License.
3

4
import { Tensor } from '../../../tensor';
5
import { ShapeUtil } from '../../../util';
6
import { WebGLInferenceHandler } from '../inference-handler';
7

8
export const reshape = (handler: WebGLInferenceHandler, inputs: Tensor[]): Tensor[] => {
9
  const reshapedDims = ShapeUtil.calculateReshapedDims(inputs[0].dims, inputs[1].integerData);
10
  if (handler.session.pack) {
11
    return [handler.reshapePacked(inputs[0], reshapedDims)];
12
  } else {
13
    return [handler.reshapeUnpacked(inputs[0], reshapedDims)];
14
  }
15
};
16