@ai-on-browser/data-analysis-models/lib/model/nns/onnx/layer/average_pool.d.ts

Data analysis model package without any dependencies

declare namespace _default {
    /**
     * Export to onnx object.
     * @param {onnx.ModelProto} model Model object
     * @param {import("../../graph.js").LayerObject & {type: 'average_pool'}} obj Node object
     * @param {{[key: string]: {type: onnx.TensorProto.DataType; size: number[]}}} info Output informatino of other layers
     * @returns {{type: onnx.TensorProto.DataType; size: number[]}} Output information of this layer
     */
    function _export(model: onnx.ModelProto, obj: {
        type: "average_pool";
        kernel: number | number[];
        stride?: number | number[];
        padding?: number | number[];
        channel_dim?: number;
    } & {
        input?: string | string[];
        name?: string;
    } & {
        type: "average_pool";
    }, info: {
        [key: string]: {
            type: onnx.TensorProto.DataType;
            size: number[];
        };
    }): {
        type: onnx.TensorProto.DataType;
        size: number[];
    };
    export { _export as export };
}
export default _default;