Contents

struct

SushiAI::Graph::BatchNormGrads

Holds the three gradients produced by batch normalisation backward.

Declared in
include/SushiAI/graph/builder.hpp

Public attributes

ValueId dx

The [N,H,W,C] input gradient.

ValueId dgamma

The [C] scale gradient.

ValueId dbeta

The [C] shift gradient.