Searched refs:BatchNormalizationTest (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc51 class BatchNormalizationTest class in namespace:xla::__anon25717
55 BatchNormalizationTest() : input_array_(kSamples, kZ, kY, kX) { function in class:xla::__anon25717::BatchNormalizationTest
86 INSTANTIATE_TEST_CASE_P(BatchNormalizationTestInstance, BatchNormalizationTest,
89 INSTANTIATE_TEST_CASE_P(BatchNormalizationTestInstance, BatchNormalizationTest,
93 XLA_TEST_P(BatchNormalizationTest, SubtractInZ) {
109 XLA_TEST_P(BatchNormalizationTest, SquareTesseractElementwise) {
126 XLA_TEST_P(BatchNormalizationTest, SumToZ) {
138 XLA_TEST_P(BatchNormalizationTest, SquareAndReduce) {
153 XLA_TEST_P(BatchNormalizationTest, VarianceToStddev) {
164 XLA_TEST_P(BatchNormalizationTest, SpecComparisonForwar
[all...]
/external/tensorflow/tensorflow/python/ops/
H A Dnn_batchnorm_test.py39 class BatchNormalizationTest(test.TestCase): class in inherits:test.TestCase
H A Dnn_fused_batchnorm_test.py34 class BatchNormalizationTest(test.TestCase): class in inherits:test.TestCase

Completed in 453 milliseconds