gx优化

chenyc

2025-05-29 92f69c57b920cf62ecc9f15f9ed196fa26dbf2ac

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import * as tf from '@tensorflow/tfjs-core';
export declare type ExtractWeightsFunction = (numWeights: number) => Float32Array;
export declare type ParamMapping = {
    originalPath?: string;
    paramPath: string;
};
export declare type ConvParams = {
    filters: tf.Tensor4D;
    bias: tf.Tensor1D;
};
export declare type FCParams = {
    weights: tf.Tensor2D;
    bias: tf.Tensor1D;
};
export declare class SeparableConvParams {
    depthwise_filter: tf.Tensor4D;
    pointwise_filter: tf.Tensor4D;
    bias: tf.Tensor1D;
    constructor(depthwise_filter: tf.Tensor4D, pointwise_filter: tf.Tensor4D, bias: tf.Tensor1D);
}