Searched refs:BuildConditionComputation (Results 1 – 2 of 2) sorted by relevance
37 std::unique_ptr<HloComputation> BuildConditionComputation( in BuildConditionComputation() function in xla::__anon9d1b4d8c0111::WhileTransformerTest128 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()139 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10)); in TEST_F()150 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5)); in TEST_F()161 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
413 std::unique_ptr<HloComputation> BuildConditionComputation( in BuildConditionComputation() function in xla::__anon1fa5445b0111::WhileCopyInsertionTest761 BuildConditionComputation(loop_state_shape)); in BuildWhileInstructionWithCustomInit()795 BuildConditionComputation(loop_state_shape_)); in TEST_F()824 BuildConditionComputation(loop_state_shape_)); in TEST_F()868 BuildConditionComputation(loop_state_shape_)); in TEST_F()884 BuildConditionComputation(loop_state_shape_)); in TEST_F()886 BuildConditionComputation(loop_state_shape_)); in TEST_F()941 BuildConditionComputation(loop_state_shape_)); in TEST_F()943 BuildConditionComputation(loop_state_shape_)); in TEST_F()1021 BuildConditionComputation(nested_loop_state_shape_)); in TEST_F()[all …]