Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc5006 std::optional<ParsedWhileLoop> parsed_while_loop = in TEST_F() local
5008 ASSERT_TRUE(parsed_while_loop.has_value()); in TEST_F()
5009 EXPECT_FALSE(parsed_while_loop->is_dynamic()); in TEST_F()
5010 EXPECT_EQ(parsed_while_loop->static_while_loop->trip_count, 5); in TEST_F()
5011 EXPECT_EQ(parsed_while_loop->static_while_loop->induction_var_index, 0); in TEST_F()
5012 EXPECT_EQ(parsed_while_loop->static_while_loop->induction_var_init_value, 0); in TEST_F()
5013 EXPECT_EQ(parsed_while_loop->static_while_loop->step_size, 1); in TEST_F()
5014 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 5); in TEST_F()
5055 std::optional<ParsedWhileLoop> parsed_while_loop = in TEST_F() local
5057 ASSERT_TRUE(parsed_while_loop.has_value()); in TEST_F()
[all …]
H A Dhlo_evaluator.cc3266 std::optional<ParsedWhileLoop> parsed_while_loop = in TryParseAndEvaluateWhileInductionVar() local
3268 if (!parsed_while_loop.has_value() || parsed_while_loop->is_dynamic()) { in TryParseAndEvaluateWhileInductionVar()
3274 parsed_while_loop->static_while_loop->induction_var_init_value + in TryParseAndEvaluateWhileInductionVar()
3275 parsed_while_loop->static_while_loop->trip_count * in TryParseAndEvaluateWhileInductionVar()
3276 parsed_while_loop->static_while_loop->step_size; in TryParseAndEvaluateWhileInductionVar()
3278 parsed_while_loop->static_while_loop->induction_var_index); in TryParseAndEvaluateWhileInductionVar()
3287 if (i == parsed_while_loop->static_while_loop->induction_var_index) { in TryParseAndEvaluateWhileInductionVar()
3306 std::optional<ParsedWhileLoop> parsed_while_loop = in HandleWhile() local
3310 if (!parsed_while_loop.has_value() || parsed_while_loop->is_dynamic() || in HandleWhile()
3312 parsed_while_loop->static_while_loop->induction_var_index != in HandleWhile()
[all …]