2020-08-20 02:05:34 +02:00
|
|
|
"use strict";
|
|
|
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
|
|
exports.disposeUnusedWeightTensors = void 0;
|
|
|
|
|
function disposeUnusedWeightTensors(weightMap, paramMappings) {
|
2020-08-18 14:04:33 +02:00
|
|
|
Object.keys(weightMap).forEach(path => {
|
|
|
|
|
if (!paramMappings.some(pm => pm.originalPath === path)) {
|
|
|
|
|
weightMap[path].dispose();
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
}
|
2020-08-20 02:05:34 +02:00
|
|
|
exports.disposeUnusedWeightTensors = disposeUnusedWeightTensors;
|
2020-08-18 14:04:33 +02:00
|
|
|
//# sourceMappingURL=disposeUnusedWeightTensors.js.map
|
