mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-01 03:45:06 +00:00
4ebc9c3b5e
* add web * add script and test * fix lint * add test/data/ops * add test/data/node/ to gitignore * modify scripts * add onnxjs * fix tests * fix test-runner * fix sourcemap * fix onnxjs profiling * update test list * update README * resolve comments * set wasm as default backend * rename package * update copyright header * do not use class "Buffer" in browser context * revise readme
15 lines
584 B
TypeScript
15 lines
584 B
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
import {Squeeze} from '../../../ops/squeeze';
|
|
import {Tensor} from '../../../tensor';
|
|
import {ShapeUtil} from '../../../util';
|
|
import {WebGLInferenceHandler} from '../inference-handler';
|
|
import {reshape} from './reshape';
|
|
|
|
export class WebGLSqueeze extends Squeeze {
|
|
run(inferenceHandler: WebGLInferenceHandler, inputs: Tensor[]): Tensor[] {
|
|
const outputDims = ShapeUtil.squeezeShape(inputs[0].dims, this.axes);
|
|
return [reshape(inferenceHandler, inputs[0], outputDims)];
|
|
}
|
|
}
|