Searched defs:BatchNormInference (Results 1 – 3 of 3) sorted by relevance
169 ## BatchNormInference section in Operation Semantics
109 TEST_F(LayoutAssignmentTest, BatchNormInference) { in TEST_F() argument
2068 XlaOp XlaBuilder::BatchNormInference(const XlaOp& operand, const XlaOp& scale, in BatchNormInference() function in xla::XlaBuilder3493 XlaOp BatchNormInference(const XlaOp& operand, const XlaOp& scale, in BatchNormInference() function