Home
last modified time | relevance | path

Searched refs:UnrollMode (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.h37 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,
Dllvm_loop.cc36 llvm::Value* step, UnrollMode unroll_mode, in ForLoop()
49 llvm::Value* step, llvm::IRBuilder<>* b, UnrollMode unroll_mode, in EmitForLoop()
150 if (unroll_mode_ == xla::llvm_ir::UnrollMode::kNoUnroll) { in GetLoopMetadata()
161 if (unroll_mode_ == xla::llvm_ir::UnrollMode::kFullyUnroll) { in GetLoopMetadata()
180 UnrollMode unroll_mode, in AddLoop()
188 llvm::Value* stride, UnrollMode unroll_mode, bool prevent_vectorization) { in AddLoop()
214 UnrollMode unroll_mode, in AddLoop()
225 UnrollMode unroll_mode, in AddLoop()
Dkernel_support_library.h38 llvm_ir::UnrollMode unroll_mode = llvm_ir::UnrollMode::kNoUnroll,
304 llvm_ir::UnrollMode unroll_mode_;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc172 KernelSupportLibrary ksl(b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitLoop()
Dir_emitter_unnested.cc4587 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitPrintfWithThreadId()
4710 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitTilingKernel()
5647 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitReductionFromOrToContiguousDimensions()
5754 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitElementForInputFusibleSlices()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc595 ? xla::llvm_ir::UnrollMode::kNoUnroll in EmitNaiveLlvmIrGemm()
596 : xla::llvm_ir::UnrollMode::kDefaultUnroll); in EmitNaiveLlvmIrGemm()