Searched refs:unroll_mode (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_loop.cc | 35 llvm::Value* step, UnrollMode unroll_mode, in ForLoop() argument
43 unroll_mode_(unroll_mode), in ForLoop()
48 llvm::Value* step, llvm::IRBuilder<>* b, UnrollMode unroll_mode, in EmitForLoop() argument
51 end_index, step, unroll_mode, in EmitForLoop()
179 UnrollMode unroll_mode, in AddLoop() argument
182 unroll_mode, prevent_vectorization); in AddLoop()
187 llvm::Value* stride, UnrollMode unroll_mode, bool prevent_vectorization) { in AddLoop() argument
193 /*prefix=*/name_, suffix, start_index, end_index, stride, unroll_mode, in AddLoop()
213 UnrollMode unroll_mode, in AddLoop() argument
217 GetConstantWithIndexType(end_index), unroll_mode, in AddLoop()
[all …]
|
| D | llvm_loop.h | 86 UnrollMode unroll_mode = llvm_ir::UnrollMode::kDefaultUnroll,
141 UnrollMode unroll_mode, bool prevent_vectorization);
199 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
206 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
214 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
220 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
|
| D | kernel_support_library.h | 38 llvm_ir::UnrollMode unroll_mode = llvm_ir::UnrollMode::kNoUnroll,
41 unroll_mode_(unroll_mode), in b_()
|