2020-08-20 02:05:34 +02:00
|
|
|
"use strict";
|
|
|
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
|
|
exports.extractParamsFromWeigthMapTiny = void 0;
|
|
|
|
|
const common_1 = require("../common");
|
|
|
|
|
const loadParamsFactory_1 = require("./loadParamsFactory");
|
|
|
|
|
function extractParamsFromWeigthMapTiny(weightMap) {
|
2020-08-18 14:04:33 +02:00
|
|
|
const paramMappings = [];
|
2020-08-20 02:05:34 +02:00
|
|
|
const { extractDenseBlock3Params } = loadParamsFactory_1.loadParamsFactory(weightMap, paramMappings);
|
2020-08-18 14:04:33 +02:00
|
|
|
const params = {
|
|
|
|
|
dense0: extractDenseBlock3Params('dense0', true),
|
|
|
|
|
dense1: extractDenseBlock3Params('dense1'),
|
|
|
|
|
dense2: extractDenseBlock3Params('dense2')
|
|
|
|
|
};
|
2020-08-20 02:05:34 +02:00
|
|
|
common_1.disposeUnusedWeightTensors(weightMap, paramMappings);
|
2020-08-18 14:04:33 +02:00
|
|
|
return { params, paramMappings };
|
|
|
|
|
}
|
2020-08-20 02:05:34 +02:00
|
|
|
exports.extractParamsFromWeigthMapTiny = extractParamsFromWeigthMapTiny;
|
2020-08-18 14:04:33 +02:00
|
|
|
//# sourceMappingURL=extractParamsFromWeigthMapTiny.js.map
|
