extractSeparableConvParamsFactory.d.ts 471 B

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