Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.h176 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.cc204 parameter_addresses_buffer = llvm_ir::EmitAllocaAtFunctionEntryWithCount( in GetArrayFunctionCallArguments()
Dir_emitter.cc592 llvm::Value* values = llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleSort()
595 llvm::Value* sizes = llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleSort()
803 llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleSelectAndScatter()
2233 llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1609 llvm::Value* sample_address = llvm_ir::EmitAllocaAtFunctionEntryWithCount( in MakePhiloxRngElementGenerator()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc843 llvm_ir::EmitAllocaAtFunctionEntryWithCount( in HandleSelectAndScatter()