Searched refs:BatchNormalizationTest (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 52 class BatchNormalizationTest : public ClientLibraryTestBase { class
54 BatchNormalizationTest() : input_array_(kSamples, kZ, kY, kX) { in BatchNormalizationTest() function in xla::__anon6fd372c60111::BatchNormalizationTest
88 XLA_TEST_F(BatchNormalizationTest, SubtractInZ) { in XLA_TEST_F() argument
104 XLA_TEST_F(BatchNormalizationTest, SquareTesseractElementwise) { in XLA_TEST_F() argument
121 XLA_TEST_F(BatchNormalizationTest, SumToZ) { in XLA_TEST_F() argument
132 XLA_TEST_F(BatchNormalizationTest, SquareAndReduce) { in XLA_TEST_F() argument
146 XLA_TEST_F(BatchNormalizationTest, VarianceToStddev) { in XLA_TEST_F() argument
157 XLA_TEST_F(BatchNormalizationTest, SpecComparisonForward) { in XLA_TEST_F() argument
216 XLA_TEST_F(BatchNormalizationTest, BasicTraining) { in XLA_TEST_F() argument
239 XLA_TEST_F(BatchNormalizationTest, BasicTraining_fp16) { in XLA_TEST_F() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_fused_batchnorm_test.py | 35 class BatchNormalizationTest(test.TestCase): class
|
| H A D | nn_batchnorm_test.py | 33 class BatchNormalizationTest(test.TestCase): class
|