| D | batch_normalization_test.cc | 56 BatchNormalizationTest() : input_array_(kSamples, kZ, kY, kX) { in BatchNormalizationTest()
68 CHECK_EQ(kZ, input_array_.depth()); in BatchNormalizationTest()
85 static constexpr int64 kZ = 2; member in xla::__anoncfc9dfd00111::BatchNormalizationTest
109 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
126 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
220 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
|