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