Searched refs:UnrollMode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_loop.h | 37 enum class UnrollMode { enum
86 UnrollMode unroll_mode = llvm_ir::UnrollMode::kDefaultUnroll,
141 UnrollMode unroll_mode, bool prevent_vectorization);
173 UnrollMode unroll_mode_;
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 | llvm_loop.cc | 35 llvm::Value* step, UnrollMode unroll_mode, in ForLoop()
48 llvm::Value* step, llvm::IRBuilder<>* b, UnrollMode unroll_mode, in EmitForLoop()
149 if (unroll_mode_ == xla::llvm_ir::UnrollMode::kNoUnroll) { in GetLoopMetadata()
160 if (unroll_mode_ == xla::llvm_ir::UnrollMode::kFullyUnroll) { in GetLoopMetadata()
179 UnrollMode unroll_mode, in AddLoop()
187 llvm::Value* stride, UnrollMode unroll_mode, bool prevent_vectorization) { in AddLoop()
213 UnrollMode unroll_mode, in AddLoop()
224 UnrollMode unroll_mode, in AddLoop()
|
| D | kernel_support_library.h | 38 llvm_ir::UnrollMode unroll_mode = llvm_ir::UnrollMode::kNoUnroll,
299 llvm_ir::UnrollMode unroll_mode_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 452 ? xla::llvm_ir::UnrollMode::kNoUnroll in EmitNaiveLlvmIrGemm()
453 : xla::llvm_ir::UnrollMode::kDefaultUnroll); in EmitNaiveLlvmIrGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3125 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitKernel()
|