21 lines
627 B
TypeScript
21 lines
627 B
TypeScript
|
import * as tf from '../../dist/tfjs.esm.js';
|
||
|
|
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);
|
||
|
|
}
|