Commit 01da2081 authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

[StreamExecutor] Allow null batch_mean/batch_var in calls to BatchNormalizationForward. 

cudnn allows these pointers to be null, so we should too.  This lets us
avoid an unnecessary ThenMemZero, and lets callers avoid an unnecessary
allocation.

PiperOrigin-RevId: 179350594
parent 6c55eb1c
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment