Commit 5ee38042 authored by Yuxin Wu's avatar Yuxin Wu Committed by Shanqing Cai
Browse files

Add kernels for FusedBatchNormGrad when is_training=False (#12580) 

* Add kernels for FusedBatchNormGrad when is_training=False. (fix #10857)

* remove unnecessary headers

* Rename some variables

* Update comments about reserve_space of FusedBatchNormGrad

* Comments and variable rename to avoid confusing of batch var/pop var/inv var, etc

* fix build for CPU

* (Hopefully) fix build for android.
parent e9d5ee1e
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