Searched refs:BuildWhileBodyComputation (Results 1 – 2 of 2) sorted by relevance
201 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anonf885129d0111::BufferDonationTest228 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in CreateTestModule()
217 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anon3198ac380111::BufferAssignmentTest1026 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F()2021 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anon3198ac380111::WhileBufferAssignmentTest2084 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2094 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2402 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2412 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2582 auto body = module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2656 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2670 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()