Searched refs:is_first_iteration (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_support_library.h | 55 bool is_first_iteration)>& for_body_generator);
60 const std::function<void(llvm::Value* ind_var, bool is_first_iteration)>& in For()
65 [&](llvm::Value* ind_var, bool is_first_iteration) -> Status { in For()
66 for_body_generator(ind_var, is_first_iteration); in For()
74 llvm::Value* ind_var, bool is_first_iteration)>& for_body_generator) { in ForWithStatus()
82 const std::function<void(llvm::Value* ind_var, bool is_first_iteration)>& in For()
107 llvm::Value* is_first_iteration)>&
113 llvm::Value* is_first_iteration)>& in For()
117 [&](llvm::Value* ind_var, llvm::Value* is_first_iteration) -> Status { in For()
118 for_body_generator(ind_var, is_first_iteration); in For()
[all …]
|
| D | kernel_support_library.cc | 41 [&](llvm::Value* indvar, bool is_first_iteration) -> Status { in ForWithStatus() argument
42 return for_body_generator(indvar, b_->getInt1(is_first_iteration)); in ForWithStatus()
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape.py | 256 def cond(grads, ready_to_update, is_first_iteration): argument
262 is_first_iteration,
270 def body(grads, ready_to_update, is_first_iteration): argument
272 del grads, ready_to_update, is_first_iteration
311 is_first_iteration = False
312 return grads, ready_to_update, is_first_iteration
|
| /external/deqp/external/openglcts/modules/glesext/tessellation_shader/ |
| D | esextcTessellationShaderVertexSpacing.cpp | 621 bool is_first_iteration = true; in getEdgesForTrianglesTessellation() local
638 if (is_first_iteration) in getEdgesForTrianglesTessellation()
645 is_first_iteration = false; in getEdgesForTrianglesTessellation()
|