extractParamsFromWeigthMapTiny.js 826 B

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