sharedLayers.js 915 B

123456789101112131415161718192021
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var tf = require("@tensorflow/tfjs-core");
  4. var common_1 = require("../common");
  5. var prelu_1 = require("./prelu");
  6. function sharedLayer(x, params, isPnet) {
  7. if (isPnet === void 0) { isPnet = false; }
  8. return tf.tidy(function () {
  9. var out = common_1.convLayer(x, params.conv1, 'valid');
  10. out = prelu_1.prelu(out, params.prelu1_alpha);
  11. out = tf.maxPool(out, isPnet ? [2, 2] : [3, 3], [2, 2], 'same');
  12. out = common_1.convLayer(out, params.conv2, 'valid');
  13. out = prelu_1.prelu(out, params.prelu2_alpha);
  14. out = isPnet ? out : tf.maxPool(out, [3, 3], [2, 2], 'valid');
  15. out = common_1.convLayer(out, params.conv3, 'valid');
  16. out = prelu_1.prelu(out, params.prelu3_alpha);
  17. return out;
  18. });
  19. }
  20. exports.sharedLayer = sharedLayer;
  21. //# sourceMappingURL=sharedLayers.js.map