Home
last modified time | relevance | path

Searched refs:loop_limit (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfor_thunk.cc26 ForThunk::ForThunk(const int64 loop_limit, in ForThunk() argument
30 loop_limit_(loop_limit), in ForThunk()
Dfor_thunk.h34 ForThunk(const int64 loop_limit,
Dir_emitter_unnested.h360 const int64 loop_limit);
Dir_emitter_unnested.cc2136 const HloInstruction* hlo, const int64 loop_limit) { in BuildForThunk() argument
2148 loop_limit, ir_emitter_body.ConsumeThunkSequence(), hlo); in BuildForThunk()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc1278 const int32 loop_limit = 100; in BM_WhileLoop() local
1284 Lt(iteration, ConstantR0<int32>(&builder, loop_limit)); in BM_WhileLoop()