Searched refs:for_body_generator (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_support_library.h | 55 bool is_first_iteration)>& for_body_generator);
61 for_body_generator) { in For() argument
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() argument
77 /*step=*/b_->getInt64(step), for_body_generator); in ForWithStatus()
83 for_body_generator) { in For() argument
86 /*step=*/b_->getInt64(step), for_body_generator); in For()
108 for_body_generator);
114 for_body_generator) { in For()
118 for_body_generator(ind_var, is_first_iteration); in For()
[all …]
|
| D | kernel_support_library.cc | 24 const std::function<Status(llvm::Value*, bool)>& for_body_generator) { in ForWithStatus() argument
26 TF_RETURN_IF_ERROR(for_body_generator(start, /*is_first_iteration=*/true)); in ForWithStatus()
29 [&](llvm::Value* iv) { return for_body_generator(iv, false); }); in ForWithStatus()
37 for_body_generator) { in ForWithStatus() argument
42 return for_body_generator(indvar, b_->getInt1(is_first_iteration)); in ForWithStatus()
51 for_body_generator(loop->GetIndVarValue(), in ForWithStatus()
|