3 lines
474 B
TypeScript
3 lines
474 B
TypeScript
import { ExtractWeightsFunction, ParamMapping, SeparableConvParams } from './types';
|
|
export declare function extractSeparableConvParamsFactory(extractWeights: ExtractWeightsFunction, paramMappings: ParamMapping[]): (channelsIn: number, channelsOut: number, mappedPrefix: string) => SeparableConvParams;
|
|
export declare function loadSeparableConvParamsFactory(extractWeightEntry: <T>(originalPath: string, paramRank: number) => T): (prefix: string) => SeparableConvParams;
|