Searched refs:ProcessUnnestedSubcomputation (Results 1 – 2 of 2) sorted by relevance
638 ProcessUnnestedSubcomputation(call->to_apply())); in HandleCall()668 ProcessUnnestedSubcomputation(xla_while->while_body())); in HandleWhile()672 ProcessUnnestedSubcomputation(xla_while->while_condition())); in HandleWhile()691 ProcessUnnestedSubcomputation(conditional->branch_computation(0))); in HandleConditional()696 ProcessUnnestedSubcomputation(conditional->branch_computation(j))); in HandleConditional()785 HloCostAnalysis::ProcessUnnestedSubcomputation(HloComputation* computation) { in ProcessUnnestedSubcomputation() function in xla::HloCostAnalysis
177 StatusOr<Properties> ProcessUnnestedSubcomputation(