"use strict";
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
Object.defineProperty(exports, "__esModule", { value: true });
exports.WebGLFlatten = void 0;
const flatten_1 = require("../../../ops/flatten");
const util_1 = require("../../../util");
const reshape_1 = require("./reshape");
class WebGLFlatten extends flatten_1.Flatten {
    run(inferenceHandler, inputs) {
        const outputDims = util_1.ShapeUtil.flattenShape(inputs[0].dims, this.axis);
        return [reshape_1.reshape(inferenceHandler, inputs[0], outputDims)];
    }
}
exports.WebGLFlatten = WebGLFlatten;
//# sourceMappingURL=flatten.js.map