Searched refs:BuildWhileConditionComputation (Results 1 – 2 of 2) sorted by relevance
178 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anon2f3f9a930111::BufferDonationTest229 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in CreateTestModule()
210 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anond1a23f2e0111::BufferAssignmentTest1043 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F()2086 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anond1a23f2e0111::WhileBufferAssignmentTest2160 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()