extractWeightEntryFactory.js 581 B

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