Searched refs:visit_function (Results 1 – 2 of 2) sorted by relevance
244 VisitStates* visit_state, const VisitFunction& visit_function, in VisitTopologicalOrder() argument279 TF_RETURN_IF_ERROR(visit_function(hlo, instruction_group)); in VisitTopologicalOrder()319 auto visit_function = in VerifyComputations() local333 VisitTopologicalOrder(&visit_states, visit_function, instruction)); in VerifyComputations()351 auto visit_function = in ComputeReachability() local361 VisitTopologicalOrder(&visit_states, visit_function, root)); in ComputeReachability()
92 const VisitFunction& visit_function,