Home
last modified time | relevance | path

Searched refs:BuildWhileBodyComputation (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbuffer_donation_test.cc204 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anon2f3f9a930111::BufferDonationTest
231 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in CreateTestModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbuffer_assignment_test.cc236 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anond1a23f2e0111::BufferAssignmentTest
1045 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F()
2100 std::unique_ptr<HloComputation> BuildWhileBodyComputation( in BuildWhileBodyComputation() function in xla::__anond1a23f2e0111::WhileBufferAssignmentTest
2162 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()