Home
last modified time | relevance | path

Searched refs:kDefaultUnroll (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.h36 kDefaultUnroll, enumerator
84 UnrollMode unroll_mode = llvm_ir::UnrollMode::kDefaultUnroll,
197 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
204 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
212 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
218 UnrollMode unroll_mode = xla::llvm_ir::UnrollMode::kDefaultUnroll,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc321 KernelSupportLibrary ksl(b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitLoop()
Dir_emitter_unnested.cc3386 KernelSupportLibrary ksl(b, llvm_ir::UnrollMode::kDefaultUnroll); in EmitXTileLoop()
3439 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitTile()
3818 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitPrintfWithThreadId()
4122 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitTilingKernel()
4978 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitUnnestedReduction()
5069 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitElementForInputFusibleSlices()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc660 : xla::llvm_ir::UnrollMode::kDefaultUnroll); in EmitNaiveLlvmIrGemm()