Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_function.cc53 num_dynamic_loop_bounds_(num_dynamic_loop_bounds) { in IrFunction()
63 DynamicLoopBounds dynamic_loop_bounds(num_dynamic_loop_bounds_); in GetDynamicLoopBounds()
64 for (int i = 0; i < num_dynamic_loop_bounds_; ++i) { in GetDynamicLoopBounds()
138 GetComputeFunctionParams(llvm_module_, num_dynamic_loop_bounds_), in Initialize()
159 if (num_dynamic_loop_bounds_ > 0) { in Initialize()
193 CHECK_GT(num_dynamic_loop_bounds_, 0); in GetDynamicLoopBound()
194 CHECK_LT(offset, num_dynamic_loop_bounds_ * 2); in GetDynamicLoopBound()
H A Dir_emitter.h543 int64_t num_dynamic_loop_bounds_ = 0; variable
549 return num_dynamic_loop_bounds_ > 0 && in ShouldEmitParallelLoopFor()
H A Dir_function.h118 int64_t num_dynamic_loop_bounds_ = 0; variable
H A Dir_emitter.cc175 num_dynamic_loop_bounds_ = 0; in EmitComputation()
177 num_dynamic_loop_bounds_ = in EmitComputation()
248 num_dynamic_loop_bounds_)); in InitializeIrFunction()
3346 TF_RET_CHECK(num_dynamic_loop_bounds_ == 0); in EmitTargetElementLoop()