Searched defs:for_body_generator (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
H A D | kernel_support_library.h | 45 // `for_body_generator(/*ind_var=*/start, /*is_first_iteration=*/true)`; 47 // `for_body_generator(/*ind_var=*/,i, /*is_first_iteration=*/false)`; 53 for_body_generator); 58 for_body_generator) { 61 /*step=*/ir_builder_->getInt64(step), for_body_generator); 68 // `for_body_generator(/*ind_var=*/start, /*is_first_iteration=*/,true)`; 70 // `for_body_generator(/*ind_var=*/,i, /*is_first_iteration=*/,false)`; 76 // `for_body_generator(/*ind_var=*/,i, 82 for_body_generator); 88 for_body_generator) { 55 For( tensorflow::StringPiece name, int64 start, int64 end, int64 step, const std::function<void(llvm::Value* ind_var, bool is_first_iteration)>& for_body_generator) argument 84 For(tensorflow::StringPiece name, llvm::Value* start, llvm::Value* end, int64 step, bool peel_first_iteration, const std::function<void(llvm::Value* ind_var, llvm::Value* is_first_iteration)>& for_body_generator) argument 94 For( tensorflow::StringPiece name, llvm::Value* start, llvm::Value* end, llvm::Value* step, const std::function<void(llvm::Value* ind_var)>& for_body_generator) argument 103 For( tensorflow::StringPiece name, int64 start, int64 end, int64 step, const std::function<void(llvm::Value* ind_var)>& for_body_generator) argument [all...] |
H A D | kernel_support_library.cc | 25 const std::function<void(llvm::Value*, bool)>& for_body_generator) { 27 for_body_generator(start, /*is_first_iteration=*/true); 29 [&](llvm::Value* iv) { for_body_generator(iv, false); }); 36 const std::function<void(llvm::Value*, llvm::Value*)>& for_body_generator) { 40 for_body_generator(indvar, ir_builder_->getInt1(is_first_iteration)); 48 for_body_generator(loop->GetIndVarValue(), 22 For( tensorflow::StringPiece name, llvm::Value* start, llvm::Value* end, llvm::Value* step, const std::function<void(llvm::Value*, bool)>& for_body_generator) argument 33 For( tensorflow::StringPiece name, llvm::Value* start, llvm::Value* end, llvm::Value* step, bool peel_first_iteration, const std::function<void(llvm::Value*, llvm::Value*)>& for_body_generator) argument
|
Completed in 243 milliseconds