Searched defs:BatchNormInference (Results 1 – 3 of 3) sorted by relevance
212 ## BatchNormInference section in Operation Semantics
111 TEST_F(LayoutAssignmentTest, BatchNormInference) { in TEST_F() argument
2032 XlaOp XlaBuilder::BatchNormInference(XlaOp operand, XlaOp scale, XlaOp offset, in BatchNormInference() function in xla::XlaBuilder3561 XlaOp BatchNormInference(const XlaOp operand, const XlaOp scale, in BatchNormInference() function