Searched refs:BuildWhileConditionComputation (Results 1 – 1 of 1) sorted by relevance
184 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anon3198ac380111::BufferAssignmentTest852 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F()1858 std::unique_ptr<HloComputation> BuildWhileConditionComputation( in BuildWhileConditionComputation() function in xla::__anon3198ac380111::WhileBufferAssignmentTest1934 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()1944 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2253 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2263 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2399 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2473 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()2487 module->AddEmbeddedComputation(BuildWhileConditionComputation("cond")); in TEST_F()