Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.h153 llvm::GlobalVariable* AllocateSharedMemoryTile(llvm::Module* module,
Dsort_util.cc336 param_shmem_buffers[i] = llvm_ir::AllocateSharedMemoryTile( in EmitSortInPlace()
Dllvm_util.cc266 llvm::GlobalVariable* AllocateSharedMemoryTile(llvm::Module* module, in AllocateSharedMemoryTile() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4310 llvm_ir::AllocateSharedMemoryTile(b_.GetInsertBlock()->getModule(), in EmitPrologueForReduction()
4857 return llvm_ir::AllocateSharedMemoryTile(b_.GetInsertBlock()->getModule(), in EmitHlo021Tile()