Searched refs:scale4D (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 505 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); in XLA_TEST_P() local
510 scale4D, offset4D, epsilon); in XLA_TEST_P()
606 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); in XLA_TEST_P() local
611 scale4D, offset4D, epsilon); in XLA_TEST_P()
725 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); in XLA_TEST_P() local
757 scale4D, rsqrt_var_add_epsilon, [](float a, float b) { return a * b; }); in XLA_TEST_P()
789 grad_activation, scale4D, [](float a, float b) { return a * b; }); in XLA_TEST_P()
|