Searched defs:BatchNormTestManySizes (Results 1 – 1 of 1) sorted by relevance
395 class BatchNormTestManySizes class399 BatchNormTestManySizes() { in BatchNormTestManySizes() function in xla::__anoncfc9dfd00111::BatchNormTestManySizes449 XLA_TEST_P(BatchNormTestManySizes, RandomizedTrainingTests) { in XLA_TEST_P() argument550 XLA_TEST_P(BatchNormTestManySizes, RandomizedInferencingTests) { in XLA_TEST_P() argument658 XLA_TEST_P(BatchNormTestManySizes, RandomizedGradTests) { in XLA_TEST_P() argument