Home
last modified time | relevance | path

Searched refs:trip_count (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_analysis.cc358 if (auto trip_count = PatternMatchLoopTripCount(while_op, *indvar_tuple_idx, in ComputeWhileLoopTripCount() local
360 return trip_count; in ComputeWhileLoopTripCount()
373 for (int64 trip_count = 0; trip_count != max_brute_force_iters + 1; in ComputeWhileLoopTripCount() local
374 ++trip_count) { in ComputeWhileLoopTripCount()
382 VLOG(2) << "Loop has static trip count of " << trip_count; in ComputeWhileLoopTripCount()
383 return trip_count; in ComputeWhileLoopTripCount()
Dwhile_loop_trip_count_annotator.cc29 if (auto trip_count = ComputeWhileLoopTripCount(instr)) { in Run() local
31 config.mutable_known_trip_count()->set_n(*trip_count); in Run()
Dwhile_util.cc153 int32 trip_count) { in MakeCountedLoopConditionComputation() argument
161 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(trip_count))); in MakeCountedLoopConditionComputation()
228 HloComputation* computation, int32 trip_count, in MakeCountedLoop() argument
232 CHECK_GE(trip_count, 0); in MakeCountedLoop()
237 MakeCountedLoopConditionComputation(loop_state_shape, trip_count)); in MakeCountedLoop()
Dwhile_util.h80 HloComputation* computation, int32 trip_count,
Dwhile_loop_simplifier.cc475 optional<int64> trip_count = in TryRemoveWhileLoop() local
478 if (trip_count && *trip_count == 0) { in TryRemoveWhileLoop()
493 if (trip_count && *trip_count == 1) { in TryRemoveWhileLoop()
/external/mesa3d/src/compiler/nir/
Dnir_opt_loop_unroll.c184 for (unsigned i = 0; i < loop->info->trip_count; i++) { in simple_unroll()
343 num_times_to_clone = loop->info->trip_count + 1; in complex_unroll()
371 num_times_to_clone = loop->info->trip_count; in complex_unroll()
475 if (li->trip_count > max_iter) in is_loop_small_enough_to_unroll()
482 li->num_instructions * li->trip_count <= max_iter * LOOP_UNROLL_LIMIT; in is_loop_small_enough_to_unroll()
550 if (loop->info->trip_count == 0 && !limiting_term_second) { in process_loops()
Dnir_loop_analyze.c629 state->loop->info->trip_count = min_trip_count; in find_trip_count()
663 state->loop->info->trip_count) { in force_unroll_array_access()
Dnir.h1635 unsigned trip_count; member
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_dependence_helpers.cpp310 SENode* trip_count = in GetFinalTripInductionNode() local
316 scalar_evolution_.CreateMultiplyNode(trip_count, induction_coefficient))); in GetFinalTripInductionNode()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_dependence_helpers.cpp310 SENode* trip_count = in GetFinalTripInductionNode() local
316 scalar_evolution_.CreateMultiplyNode(trip_count, induction_coefficient))); in GetFinalTripInductionNode()
/external/swiftshader/third_party/llvm-7.0/llvm/test/Transforms/LoopVectorize/
Dtripcount.ll36 ; trip_count per invocation and not worth to vectorize.