Home
last modified time | relevance | path

Searched defs:ksl (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc114 KernelSupportLibrary ksl(b); in EmitCompareLoopBody() local
162 KernelSupportLibrary ksl(b); in EmitTiledCompareLoop() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc321 KernelSupportLibrary ksl(b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitLoop() local
Dir_emitter_unnested.cc3386 KernelSupportLibrary ksl(b, llvm_ir::UnrollMode::kDefaultUnroll); in EmitXTileLoop() local
3439 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitTile() local
3818 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitPrintfWithThreadId() local
3868 KernelSupportLibrary ksl(&b_); in EmitReductionOutputForRowReduction() local
3966 KernelSupportLibrary ksl(&b_); in EmitReductionOutputForColumnReduction() local
4122 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitTilingKernel() local
4759 KernelSupportLibrary ksl(&b_); in EmitIRForReduction() local
4978 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitUnnestedReduction() local
5069 KernelSupportLibrary ksl(&b_, llvm_ir::UnrollMode::kDefaultUnroll); in EmitElementForInputFusibleSlices() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc1420 KernelSupportLibrary ksl(b); in EmitBatchDotOperation() local