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
290 ## BatchNormTraining section in Operation Semantics
182 TEST_F(LayoutAssignmentTest, BatchNormTraining) { in TEST_F() argument
2641 XlaOp XlaBuilder::BatchNormTraining(XlaOp operand, XlaOp scale, XlaOp offset, in BatchNormTraining() function in xla::XlaBuilder4939 XlaOp BatchNormTraining(const XlaOp operand, const XlaOp scale, in BatchNormTraining() function