Searched defs:for_body_generator (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dkernel_support_library.h45 // `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 Dkernel_support_library.cc25 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