extractParamsFromWeigthMap.js 862 B

123456789101112131415161718
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var common_1 = require("../common");
  4. var loadParamsFactory_1 = require("./loadParamsFactory");
  5. function extractParamsFromWeigthMap(weightMap) {
  6. var paramMappings = [];
  7. var extractDenseBlock4Params = loadParamsFactory_1.loadParamsFactory(weightMap, paramMappings).extractDenseBlock4Params;
  8. var params = {
  9. dense0: extractDenseBlock4Params('dense0', true),
  10. dense1: extractDenseBlock4Params('dense1'),
  11. dense2: extractDenseBlock4Params('dense2'),
  12. dense3: extractDenseBlock4Params('dense3')
  13. };
  14. common_1.disposeUnusedWeightTensors(weightMap, paramMappings);
  15. return { params: params, paramMappings: paramMappings };
  16. }
  17. exports.extractParamsFromWeigthMap = extractParamsFromWeigthMap;
  18. //# sourceMappingURL=extractParamsFromWeigthMap.js.map