aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h')
-rw-r--r--tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h11
1 files changed, 5 insertions, 6 deletions
diff --git a/tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h b/tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h
index 6be1c2fba2..c4f5c82086 100644
--- a/tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h
+++ b/tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h
@@ -41,11 +41,11 @@ class LoopEmitter {
using BodyEmitter = std::function<Status(const IrArray::Index& index)>;
LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape,
- llvm::IRBuilder<>* ir_builder);
+ llvm::IRBuilder<>* b);
// Constructs a LoopEmitter from an element generator that generates each
// element of the given target array.
LoopEmitter(const ElementGenerator& target_element_generator,
- const IrArray& target_array, llvm::IRBuilder<>* ir_builder);
+ const IrArray& target_array, llvm::IRBuilder<>* b);
// Constructs a LoopEmitter that emits one element into each of N separate
// arrays on each iteration of the loop.
@@ -54,7 +54,7 @@ class LoopEmitter {
// produce an LLVM struct with N elements.
LoopEmitter(const ElementGenerator& target_element_generator,
tensorflow::gtl::ArraySlice<IrArray> target_arrays,
- llvm::IRBuilder<>* ir_builder);
+ llvm::IRBuilder<>* b);
LoopEmitter(const LoopEmitter&) = delete;
LoopEmitter& operator=(const LoopEmitter&) = delete;
@@ -65,8 +65,7 @@ class LoopEmitter {
// specifies the element, will return multiple indices if the loop is
// unrolled.
std::vector<IrArray::Index> EmitIndexAndSetExitBasicBlock() {
- return EmitIndexAndSetExitBasicBlock(/*loop_name=*/"",
- ir_builder_->getInt64Ty());
+ return EmitIndexAndSetExitBasicBlock(/*loop_name=*/"", b_->getInt64Ty());
}
virtual std::vector<IrArray::Index> EmitIndexAndSetExitBasicBlock(
@@ -87,7 +86,7 @@ class LoopEmitter {
// scalar, no loops are emitted and exit_bb_ is nullptr in that case.
llvm::BasicBlock* exit_bb_;
- llvm::IRBuilder<>* ir_builder_;
+ llvm::IRBuilder<>* b_;
};
} // namespace llvm_ir