1
2
3
4
5
6
7
8
9
10
11
12
| import { isTensor } from '../utils';
| export function extractWeightEntryFactory(weightMap, paramMappings) {
| return function (originalPath, paramRank, mappedPath) {
| var tensor = weightMap[originalPath];
| if (!isTensor(tensor, paramRank)) {
| throw new Error("expected weightMap[" + originalPath + "] to be a Tensor" + paramRank + "D, instead have " + tensor);
| }
| paramMappings.push({ originalPath: originalPath, paramPath: mappedPath || originalPath });
| return tensor;
| };
| }
| //# sourceMappingURL=extractWeightEntryFactory.js.map
|
|
