Searched refs:grad_scale_before_reduction (Results 1 – 1 of 1) sorted by relevance
744 auto grad_scale_before_reduction = *ReferenceUtil::MapArray4D( in XLA_TEST_P() local749 grad_scale_before_reduction, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()