Searched refs:grad_output_array (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 666 Array4D<float> grad_output_array(bounds[0], bounds[1], bounds[2], bounds[3]); in XLA_TEST_P() local
667 grad_output_array.FillRandom(GetParam().random_value_var, in XLA_TEST_P()
734 *ReferenceUtil::MapArray4D(grad_output_array, var_add_epsilon, in XLA_TEST_P()
741 *ReferenceUtil::MapArray4D(grad_output_array, activation_shifted, in XLA_TEST_P()
753 ReferenceUtil::Reduce4DTo1D(grad_output_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
760 grad_output_array, [&](float a) { return num_elements_per_feature * a; }); in XLA_TEST_P()
807 LiteralUtil::CreateR4FromArray4D<float>(grad_output_array); in XLA_TEST_P()
|