Searched defs:trip_count (Results 1 – 13 of 13) sorted by relevance
31 if (auto trip_count = ComputeWhileLoopTripCount(instr)) { in Run() local
159 int32_t trip_count) { in MakeCountedLoopConditionComputation()246 HloModule* module, int32_t trip_count, in MakeCountedLoop()288 HloComputation* computation, int32_t trip_count, in MakeCountedLoop()
511 if (auto trip_count = PatternMatchLoopTripCount(while_op, *indvar_tuple_idx, in ComputeWhileLoopTripCount() local526 for (int64_t trip_count = 0; trip_count != max_brute_force_iters + 1; in ComputeWhileLoopTripCount() local
49 int64_t trip_count = -1; member
281 int64_t trip_count = 1; in GatherLoopTripCount() local
616 int64_t trip_count = in PatternMatchParseWhileLoop() local638 int trip_count = in PatternMatchParseWhileLoop() local668 int trip_count = in PatternMatchParseWhileLoop() local689 int trip_count = in PatternMatchParseWhileLoop() local3318 int64_t trip_count = parsed_while_loop->static_while_loop->trip_count; in HandleWhile() local
798 optional<int64_t> trip_count = in TryRemoveWhileLoop() local
582 unsigned trip_count) in is_access_out_of_bounds()621 unsigned trip_count) in remove_out_of_bounds_induction_use()683 partial_unroll(nir_shader *shader, nir_loop *loop, unsigned trip_count) in partial_unroll()767 unsigned trip_count = in check_unrolling_restrictions() local
63 uint64_t trip_count = (upper_bound - lower_bound).sdiv(step).getZExtValue(); in matchAndRewrite() local
310 SENode* trip_count = in GetFinalTripInductionNode() local
308 SENode* trip_count = in GetFinalTripInductionNode() local
1408 mlir::IntegerAttr trip_count; in EmitWhileOp() local