Searched defs:BatchNormTraining (Results 1 – 4 of 4) sorted by relevance
79 XLA_TEST_F(Bfloat16Test, DISABLED_ON_INTERPRETER(BatchNormTraining)) { in XLA_TEST_F()
55 TEST_F(BatchNormExpanderTest, BatchNormTraining) { in TEST_F() argument
290 ## BatchNormTraining section in Operation Semantics
2795 XlaOp XlaBuilder::BatchNormTraining(XlaOp operand, XlaOp scale, XlaOp offset, in BatchNormTraining() function in xla::XlaBuilder4973 XlaOp BatchNormTraining(const XlaOp operand, const XlaOp scale, in BatchNormTraining() function