Home
last modified time | relevance | path

Searched refs:static_while_loop (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc5010 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()
5059 EXPECT_EQ(parsed_while_loop->static_while_loop->trip_count, 10); in TEST_F()
5060 EXPECT_EQ(parsed_while_loop->static_while_loop->induction_var_index, 0); in TEST_F()
5061 EXPECT_EQ(parsed_while_loop->static_while_loop->induction_var_init_value, 0); in TEST_F()
5062 EXPECT_EQ(parsed_while_loop->static_while_loop->step_size, 1); in TEST_F()
5063 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 10); in TEST_F()
[all …]
H A Dhlo_evaluator.h66 std::optional<ParsedStaticWhileLoop> static_while_loop; member
67 bool is_dynamic() const { return !static_while_loop.has_value(); } in is_dynamic()
H A Dhlo_evaluator.cc3274 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()
3312 parsed_while_loop->static_while_loop->induction_var_index != in HandleWhile()
3318 int64_t trip_count = parsed_while_loop->static_while_loop->trip_count; in HandleWhile()