Searched refs:BuildWhileBodyComputation (Results 1 – 2 of 2) sorted by relevance
204 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anon2f3f9a930111::BufferDonationTest231 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in CreateTestModule()
236 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anond1a23f2e0111::BufferAssignmentTest1045 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F()2100 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anond1a23f2e0111::WhileBufferAssignmentTest2162 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2172 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2480 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2490 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2721 auto body = module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2794 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2808 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()