convLayer.d.ts 350 B

12345
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { ConvLayerParams } from './types';
  3. export declare function conv(x: tf.Tensor4D, params: ConvLayerParams): tf.Tensor4D;
  4. export declare function convNoRelu(x: tf.Tensor4D, params: ConvLayerParams): tf.Tensor4D;
  5. export declare function convDown(x: tf.Tensor4D, params: ConvLayerParams): tf.Tensor4D;