8 lines
269 B
TypeScript
8 lines
269 B
TypeScript
|
import * as tf from '@tensorflow/tfjs-core';
|
||
|
|
import { ParamMapping } from '../common';
|
||
|
|
import { NetParams } from './types';
|
||
|
|
export declare function extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
|
||
|
|
params: NetParams;
|
||
|
|
paramMappings: ParamMapping[];
|
||
|
|
};
|