Home
last modified time | relevance | path

Searched refs:EmitAllocaAtFunctionEntry (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc560 llvm::Value* accum_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleDot()
784 llvm::Value* return_buffer = llvm_ir::EmitAllocaAtFunctionEntry( in ComputeNestedElement()
790 parameter_buffers.push_back(llvm_ir::EmitAllocaAtFunctionEntry( in ComputeNestedElement()
Delemental_ir_emitter.cc362 llvm::Value* accum_ptr = llvm_ir::EmitAllocaAtFunctionEntry( in MakeElementGenerator()
Dir_emitter_unnested.cc838 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter()
846 llvm::Value* initialized_flag_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter()
914 llvm::Value* select_return_buffer = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc650 llvm::Value* accumulator_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalReduceWindow()
798 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter()
805 llvm::Value* initialized_flag_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter()
974 llvm::Value* sum_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalConvolution()
1574 accumulator.push_back(llvm_ir::EmitAllocaAtFunctionEntry( in EmitInnerLoopForVectorizedReduction()
1824 llvm::AllocaInst* accumulator_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalReduce()
2736 llvm_ir::EmitAllocaAtFunctionEntry(b->getInt32Ty(), "rdtscp_aux", b); in ReadCycleCounter()
2882 llvm::Value* buffer = llvm_ir::EmitAllocaAtFunctionEntry( in EmitThreadLocalBufferPointer()
3056 llvm::Value* parameter_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitThreadLocalCall()
3062 llvm::Value* return_value_buffer = llvm_ir::EmitAllocaAtFunctionEntry( in EmitThreadLocalCall()
Dvector_support_library.cc408 alloca_ = llvm_ir::EmitAllocaAtFunctionEntry(type, "", b_); in LlvmVariable()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.h169 llvm::AllocaInst* EmitAllocaAtFunctionEntry(llvm::Type* type,
Dsort_util.cc120 llvm::Value* compare_return_buffer = llvm_ir::EmitAllocaAtFunctionEntry( in EmitCompareLoopBody()
Dllvm_util.cc278 llvm::AllocaInst* EmitAllocaAtFunctionEntry(llvm::Type* type, in EmitAllocaAtFunctionEntry() function
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc937 llvm_ir::EmitAllocaAtFunctionEntry(x->getType(), "p.addr", b_); in EmitErfInv()
2021 llvm::Value* ret_value_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalDynamicUpdateSlice()
2085 llvm::Value* ret_value_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalPad()
2141 llvm_ir::EmitAllocaAtFunctionEntry(primitive_type_llvm, "dot_acc", b_); in EmitElementalDot()