import * as tf from '@tensorflow/tfjs-core';

import { ConvLayerParams } from './convLayer';
import { ResidualLayerParams } from './residualLayer';

export type FaceRecognitionNetParams = {
  conv32_down: ConvLayerParams
  conv32_1: ResidualLayerParams
  conv32_2: ResidualLayerParams
  conv32_3: ResidualLayerParams
  conv64_down: ResidualLayerParams
  conv64_1: ResidualLayerParams
  conv64_2: ResidualLayerParams
  conv64_3: ResidualLayerParams
  conv128_down: ResidualLayerParams
  conv128_1: ResidualLayerParams
  conv128_2: ResidualLayerParams
  conv256_down: ResidualLayerParams
  conv256_1: ResidualLayerParams
  conv256_2: ResidualLayerParams
  conv256_down_out: ResidualLayerParams
  fc: tf.Tensor2D
}