7 lines
576 B
TypeScript
7 lines
576 B
TypeScript
import { ExtractWeightsFunction, ParamMapping } from '../common';
|
|
import { DenseBlock3Params, DenseBlock4Params } from './types';
|
|
export declare function extractorsFactory(extractWeights: ExtractWeightsFunction, paramMappings: ParamMapping[]): {
|
|
extractDenseBlock3Params: (channelsIn: number, channelsOut: number, mappedPrefix: string, isFirstLayer?: boolean) => DenseBlock3Params;
|
|
extractDenseBlock4Params: (channelsIn: number, channelsOut: number, mappedPrefix: string, isFirstLayer?: boolean) => DenseBlock4Params;
|
|
};
|
|
//# sourceMappingURL=extractorsFactory.d.ts.map
|