Skip to content

applications/src/kernels/batchnorm.h

Functions

Name
void batchnorm_fp64(double * ifmap, double * gamma, double * beta, double * ofmap, uint32_t OW, uint32_t CI, uint32_t compute_num, uint32_t setup_SSR)
implementation of a FP64 batchnorm as a linear combination y = gamma * x + beta

Functions Documentation

function batchnorm_fp64

void batchnorm_fp64(
    double * ifmap,
    double * gamma,
    double * beta,
    double * ofmap,
    uint32_t OW,
    uint32_t CI,
    uint32_t compute_num,
    uint32_t setup_SSR
)

implementation of a FP64 batchnorm as a linear combination y = gamma * x + beta

Parameters:

  • ifmap pointer to input feature map
  • gamma pointer to gamma
  • beta pointer to beta
  • ofmap pointer to output feature map
  • OW width of output feature map
  • CI number of input channels
  • compute_num number of compute units
  • setup_SSR setup SSR strides and bounds

Source code

// Copyright 2020 ETH Zurich and University of Bologna.
// Licensed under the Apache License, Version 2.0, see LICENSE for details.
// SPDX-License-Identifier: Apache-2.0

#include "snrt.h"

void batchnorm_fp64(double *ifmap, double *gamma, double *beta, double *ofmap,
                    uint32_t OW, uint32_t CI, uint32_t compute_num,
                    uint32_t setup_SSR);

Updated on 2023-06-19 at 09:43:56 +0000