Searched defs:body_builder (Results 1 – 14 of 14) sorted by relevance
62 std::unique_ptr<XlaBuilder> body_builder = in WhileLoopHelper() local94 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in ForEachIndex()
282 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in HouseHolderBidiagonalization()621 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in WhileLoopFn()
242 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in WhileLoopFn()
339 auto body_builder = HloComputation::Builder("body"); in TEST_F() local527 auto body_builder = HloComputation::Builder("body"); in TEST_F() local622 auto body_builder = HloComputation::Builder("body"); in TEST_F() local793 auto body_builder = HloComputation::Builder("body"); in TEST_F() local943 auto body_builder = HloComputation::Builder("body"); in TEST_F() local1060 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
132 auto body_builder = HloComputation::Builder("body"); in TEST_F() local215 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
70 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in CholeskyUnblocked()
1303 auto body_builder = HloComputation::Builder("body"); in TEST_F() local1585 auto body_builder = HloComputation::Builder("body"); in TEST_F() local1647 auto body_builder = HloComputation::Builder("body"); in TEST_F() local1724 auto body_builder = HloComputation::Builder("body"); in TEST_F() local1798 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
264 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
62 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local154 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local712 auto body_builder = HloComputation::Builder("WhileBody"); in TEST_F() local
233 auto body_builder = HloComputation::Builder(TestName() + "-body"); in TEST_F() local
486 auto body_builder = HloComputation::Builder("body"); in TEST_P() local589 auto body_builder = HloComputation::Builder("body"); in TEST_P() local779 auto body_builder = HloComputation::Builder("body"); in TEST_P() local1026 auto body_builder = HloComputation::Builder("body"); in TEST_P() local1331 auto body_builder = HloComputation::Builder(TestName()); in TEST_P() local
474 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
73 auto body_builder = HloComputation::Builder("body"); in TEST_F() local
52 XlaBuilder body_builder("loop_body"); in XLA_TEST_F() local