Searched refs:embedded_computation (Results 1 – 3 of 3) sorted by relevance
761 for (HloComputation* embedded_computation : in MakeComputationPostOrder()763 if (!added_computations.contains(embedded_computation)) { in MakeComputationPostOrder()764 post_order.push_back(embedded_computation); in MakeComputationPostOrder()765 added_computations.insert(embedded_computation); in MakeComputationPostOrder()
1585 auto embedded_computation = in TEST_P() local1596 HloInstruction::CreateCall(vector_shape_, {exp}, embedded_computation)); in TEST_P()
126 XlaComputation CreateMapPlusN(const XlaComputation& embedded_computation, in CreateMapPlusN() argument130 auto map = Map(&builder, {x}, embedded_computation, {}); in CreateMapPlusN()