Searched refs:var4D (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 504 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P() local
509 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D, in XLA_TEST_P()
605 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P() local
610 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D, in XLA_TEST_P()
724 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P() local
728 var4D, [epsilon](float a) { return a + epsilon; }); in XLA_TEST_P()
|