Searched defs:BatchNormTestManySizes (Results 1 – 1 of 1) sorted by relevance
470 class BatchNormTestManySizes class512 XLA_TEST_P(BatchNormTestManySizes, RandomizedTrainingTests) { in XLA_TEST_P() argument613 XLA_TEST_P(BatchNormTestManySizes, RandomizedInferencingTests) { in XLA_TEST_P() argument721 XLA_TEST_P(BatchNormTestManySizes, RandomizedGradTests) { in XLA_TEST_P() argument