Searched refs:BuildWhileConditionComputation (Results 1 – 2 of 2) sorted by relevance
175 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anonf885129d0111::BufferDonationTest226 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in CreateTestModule()
191 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anon3198ac380111::BufferAssignmentTest1024 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F()2007 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anon3198ac380111::WhileBufferAssignmentTest2082 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2092 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2400 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2410 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2581 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2654 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2668 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()