Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc52 num_dynamic_loop_bounds_(num_dynamic_loop_bounds) { in IrFunction()
62 DynamicLoopBounds dynamic_loop_bounds(num_dynamic_loop_bounds_); in GetDynamicLoopBounds()
63 for (int i = 0; i < num_dynamic_loop_bounds_; ++i) { in GetDynamicLoopBounds()
137 GetComputeFunctionParams(llvm_module_, num_dynamic_loop_bounds_), in Initialize()
156 if (num_dynamic_loop_bounds_ > 0) { in Initialize()
182 CHECK_GT(num_dynamic_loop_bounds_, 0); in GetDynamicLoopBound()
183 CHECK_LT(offset, num_dynamic_loop_bounds_ * 2); in GetDynamicLoopBound()
Dir_function.h105 int64 num_dynamic_loop_bounds_ = 0; variable
Dir_emitter.h483 int64 num_dynamic_loop_bounds_ = 0; variable
489 return num_dynamic_loop_bounds_ > 0 && in ShouldEmitParallelLoopFor()
Dir_emitter.cc161 num_dynamic_loop_bounds_ = 0; in EmitComputation()
163 num_dynamic_loop_bounds_ = in EmitComputation()
221 num_dynamic_loop_bounds_)); in InitializeIrFunction()
3143 TF_RET_CHECK(num_dynamic_loop_bounds_ == 0); in EmitTargetElementLoop()