Searched defs:BatchNormTraining (Results 1 – 5 of 5) sorted by relevance
81 XLA_TEST_F(Bfloat16Test, DISABLED_ON_INTERPRETER(BatchNormTraining)) { in XLA_TEST_F()
56 TEST_F(BatchNormExpanderTest, BatchNormTraining) { in TEST_F() argument
204 ## BatchNormTraining section in Operation Semantics
180 TEST_F(LayoutAssignmentTest, BatchNormTraining) { in TEST_F() argument
2045 XlaOp XlaBuilder::BatchNormTraining(const XlaOp& operand, const XlaOp& scale, in BatchNormTraining() function in xla::XlaBuilder3486 XlaOp BatchNormTraining(const XlaOp& operand, const XlaOp& scale, in BatchNormTraining() function