Searched defs:grad_scale_before_reduction (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc734 auto grad_scale_before_reduction = *ReferenceUtil::MapArray4D( local
739 grad_scale_before_reduction, /*init=*/0.0f, reduce_dims,

Completed in 136 milliseconds