Home
last modified time | relevance | path

Searched refs:EmitAllocaAtFunctionEntryWithCount (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.h171 llvm::AllocaInst* EmitAllocaAtFunctionEntryWithCount(llvm::Type* type,
Dllvm_util.cc282 return EmitAllocaAtFunctionEntryWithCount(type, nullptr, name, b, alignment); in EmitAllocaAtFunctionEntry()
285 llvm::AllocaInst* EmitAllocaAtFunctionEntryWithCount(llvm::Type* type, in EmitAllocaAtFunctionEntryWithCount() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc197 arguments_buffer = llvm_ir::EmitAllocaAtFunctionEntryWithCount( in EncodeArrayFunctionArguments()
Dir_emitter.cc602 llvm::Value* values = llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleSort()
605 llvm::Value* sizes = llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleSort()
722 llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleSelectAndScatter()
2308 llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2275 llvm_ir::EmitAllocaAtFunctionEntryWithCount( in EmitSelectAndScatterFromMlir()
4258 llvm_ir::EmitAllocaAtFunctionEntryWithCount( in EmitPrologueForReduction()