Home
last modified time | relevance | path

Searched refs:loop_bound (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening.cc121 int loop_bound = GetLoopBound(while_hlo, default_loop_count, max_loop_count); in GetLoopBoundWithOuterLoopMax() local
122 if (loop_bound > max_outer_loop_count) { in GetLoopBoundWithOuterLoopMax()
129 return loop_bound; in GetLoopBoundWithOuterLoopMax()
186 const int loop_bound = GetLoopBoundWithOuterLoopMax( in FlattenWhileLoop() local
190 VLOG(1) << "loop_bound = " << loop_bound; in FlattenWhileLoop()
193 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int>(loop_bound))); in FlattenWhileLoop()
/external/tensorflow/tensorflow/python/ops/
Drnn.py946 loop_bound = math_ops.minimum(time_steps,
950 loop_bound = time_steps
953 cond=lambda time, *_: time < loop_bound,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h57 int64_t loop_bound = -1; member
Dhlo_evaluator_test.cc5014 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 5); in TEST_F()
5063 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 10); in TEST_F()
5114 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 40); in TEST_F()
5165 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 40); in TEST_F()
5222 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 10); in TEST_F()
5287 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 10); in TEST_F()
5374 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 10); in TEST_F()
5462 EXPECT_EQ(parsed_while_loop->static_while_loop->loop_bound, 1); in TEST_F()