Searched refs:ArrayOrZeros (Results 1 – 1 of 1) sorted by relevance
103 Array ArrayOrZeros(const absl::optional<Array>& array, const Array& zeros_template, Dtype dtype) { in ArrayOrZeros() function354 Array ggx = ArrayOrZeros(bctx2.output_grad(0), x, interm_dtype); in BatchNorm()355 … Array gggamma = ArrayOrZeros(bctx2.output_grad(1), gamma_reshaped, interm_dtype); in BatchNorm()356 … Array ggbeta = ArrayOrZeros(bctx2.output_grad(2), gamma_reshaped, interm_dtype); in BatchNorm()