import * as tf from '../../dist/tfjs.esm';
import { ConvParams } from '../common/index';
import { SeparableConvParams } from '../common/types';
export type BatchNorm = {
    sub: tf.Tensor1D;
    truediv: tf.Tensor1D;
};
export type ConvWithBatchNorm = {
    conv: ConvParams;
    bn: BatchNorm;
};
export type MobilenetParams = {
    conv0: SeparableConvParams | ConvParams;
    conv1: SeparableConvParams;
    conv2: SeparableConvParams;
    conv3: SeparableConvParams;
    conv4: SeparableConvParams;
    conv5: SeparableConvParams;
    conv6?: SeparableConvParams;
    conv7?: SeparableConvParams;
    conv8: ConvParams;
};
export type DefaultTinyYolov2NetParams = {
    conv0: ConvWithBatchNorm;
    conv1: ConvWithBatchNorm;
    conv2: ConvWithBatchNorm;
    conv3: ConvWithBatchNorm;
    conv4: ConvWithBatchNorm;
    conv5: ConvWithBatchNorm;
    conv6: ConvWithBatchNorm;
    conv7: ConvWithBatchNorm;
    conv8: ConvParams;
};
export type TinyYolov2NetParams = DefaultTinyYolov2NetParams | MobilenetParams;