Searched refs:BuildWhileBodyComputation (Results 1 – 1 of 1) sorted by relevance
210 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anon3198ac380111::BufferAssignmentTest854 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F()1872 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anon3198ac380111::WhileBufferAssignmentTest1936 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()1946 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2255 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2265 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2400 auto body = module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2475 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()2489 module->AddEmbeddedComputation(BuildWhileBodyComputation("body")); in TEST_F()