Searched defs:BatchNormExpanderVisitor (Results 1 – 1 of 1) sorted by relevance
49 class BatchNormExpanderVisitor : public DfsHloVisitorWithDefault { class72 explicit BatchNormExpanderVisitor(HloComputation* computation, in BatchNormExpanderVisitor() function in xla::__anon262de65b0111::BatchNormExpanderVisitor