Searched refs:MapInlinerVisitor (Results 1 – 1 of 1) sorted by relevance
36 class MapInlinerVisitor : public DfsHloVisitorWithDefault { class38 explicit MapInlinerVisitor(HloComputation* computation) in MapInlinerVisitor() function in xla::MapInlinerVisitor59 StatusOr<bool> MapInlinerVisitor::Run(HloComputation* computation) { in Run()66 Status MapInlinerVisitor::HandleMap(HloInstruction* map) { in HandleMap()115 MapInlinerVisitor visitor(/*computation=*/nullptr); in Run()