33 lines
1017 B
TypeScript
33 lines
1017 B
TypeScript
import * as tf from '@tensorflow/tfjs-core';
|
|
import { ConvParams } from '../common';
|
|
export declare type ScaleLayerParams = {
|
|
weights: tf.Tensor1D;
|
|
biases: tf.Tensor1D;
|
|
};
|
|
export declare type ResidualLayerParams = {
|
|
conv1: ConvLayerParams;
|
|
conv2: ConvLayerParams;
|
|
};
|
|
export declare type ConvLayerParams = {
|
|
conv: ConvParams;
|
|
scale: ScaleLayerParams;
|
|
};
|
|
export declare type NetParams = {
|
|
conv32_down: ConvLayerParams;
|
|
conv32_1: ResidualLayerParams;
|
|
conv32_2: ResidualLayerParams;
|
|
conv32_3: ResidualLayerParams;
|
|
conv64_down: ResidualLayerParams;
|
|
conv64_1: ResidualLayerParams;
|
|
conv64_2: ResidualLayerParams;
|
|
conv64_3: ResidualLayerParams;
|
|
conv128_down: ResidualLayerParams;
|
|
conv128_1: ResidualLayerParams;
|
|
conv128_2: ResidualLayerParams;
|
|
conv256_down: ResidualLayerParams;
|
|
conv256_1: ResidualLayerParams;
|
|
conv256_2: ResidualLayerParams;
|
|
conv256_down_out: ResidualLayerParams;
|
|
fc: tf.Tensor2D;
|
|
};
|
|
//# sourceMappingURL=types.d.ts.map
|