Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbuffer_donation_test.cc178 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anon2f3f9a930111::BufferDonationTest
229 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in CreateTestModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbuffer_assignment_test.cc210 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anond1a23f2e0111::BufferAssignmentTest
1043 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F()
2086 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anond1a23f2e0111::WhileBufferAssignmentTest
2160 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()
2170 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()
2478 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()
2488 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()
2720 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()
2792 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()
2806 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()