1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| import * as tf from '../../dist/tfjs.esm';
|
| import { SeparableConvParams } from './types';
|
| export function depthwiseSeparableConv(
| x: tf.Tensor4D,
| params: SeparableConvParams,
| stride: [number, number],
| ): tf.Tensor4D {
| return tf.tidy(() => {
| let out = tf.separableConv2d(x, params.depthwise_filter, params.pointwise_filter, stride, 'same');
| out = tf.add(out, params.bias);
| return out;
| });
| }
|
|
