| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_trip_count_annotator.cc | 31 if (auto trip_count = ComputeWhileLoopTripCount(instr)) { in Run() local
|
| D | while_util.cc | 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()
|
| D | while_loop_analysis.cc | 511 if (auto trip_count = PatternMatchLoopTripCount(while_op, *indvar_tuple_idx, in ComputeWhileLoopTripCount() local
526 for (int64_t trip_count = 0; trip_count != max_brute_force_iters + 1; in ComputeWhileLoopTripCount() local
|
| D | hlo_evaluator.h | 49 int64_t trip_count = -1; member
|
| D | gather_expander.cc | 281 int64_t trip_count = 1; in GatherLoopTripCount() local
|
| D | hlo_evaluator.cc | 616 int64_t trip_count = in PatternMatchParseWhileLoop() local
638 int trip_count = in PatternMatchParseWhileLoop() local
668 int trip_count = in PatternMatchParseWhileLoop() local
689 int trip_count = in PatternMatchParseWhileLoop() local
3318 int64_t trip_count = parsed_while_loop->static_while_loop->trip_count; in HandleWhile() local
|
| D | while_loop_simplifier.cc | 798 optional<int64_t> trip_count = in TryRemoveWhileLoop() local
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_opt_loop_unroll.c | 612 unsigned trip_count) in is_access_out_of_bounds()
662 unsigned trip_count) in remove_out_of_bounds_induction_use()
753 partial_unroll(nir_shader *shader, nir_loop *loop, unsigned trip_count) in partial_unroll()
912 unsigned trip_count = in check_unrolling_restrictions() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | canonicalize.cc | 63 uint64_t trip_count = (upper_bound - lower_bound).sdiv(step).getZExtValue(); in matchAndRewrite() local
|
| /external/pytorch/torch/csrc/jit/mobile/ |
| D | interpreter.cpp | 241 int64_t trip_count = sframe[0].toInt(); in run() local
|
| /external/pytorch/torch/csrc/jit/ir/ |
| D | ir_views.h | 118 auto trip_count = toIValue(maxTripCount()); in loopType() local
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | constant_propagation.cpp | 221 Value* trip_count = node->inputs().at(0); in loopWillNotRun() local
|
| D | loop_unrolling.cpp | 179 Value* trip_count = loop->inputs().at(0); in unroll() local
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | loop_dependence_helpers.cpp | 308 SENode* trip_count = in GetFinalTripInductionNode() local
|
| /external/angle/third_party/spirv-tools/src/source/opt/ |
| D | loop_dependence_helpers.cpp | 308 SENode* trip_count = in GetFinalTripInductionNode() local
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | loop_dependence_helpers.cpp | 308 SENode* trip_count = in GetFinalTripInductionNode() local
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | interpreter.cpp | 427 int64_t trip_count = fr[0].toInt(); in runTemplate() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1408 mlir::IntegerAttr trip_count; in EmitWhileOp() local
|
| /external/pytorch/torch/csrc/jit/frontend/ |
| D | ir_emitter.cpp | 2381 Value* trip_count = in emitLoopCommon() local
|