pointwiseConvLayer.js 351 B

123456789
  1. import * as tf from '@tensorflow/tfjs-core';
  2. export function pointwiseConvLayer(x, params, strides) {
  3. return tf.tidy(function () {
  4. var out = tf.conv2d(x, params.filters, strides, 'same');
  5. out = tf.add(out, params.batch_norm_offset);
  6. return tf.clipByValue(out, 0, 6);
  7. });
  8. }
  9. //# sourceMappingURL=pointwiseConvLayer.js.map