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