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.cc410 llvm::Value* cas_old_output_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitAtomicOperationUsingCAS()
412 llvm::Value* cas_new_output_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitAtomicOperationUsingCAS()
639 llvm::Value* return_buffer = llvm_ir::EmitAllocaAtFunctionEntry( in ComputeNestedElement()
643 parameter_buffers.push_back(llvm_ir::EmitAllocaAtFunctionEntry( in ComputeNestedElement()
Dhlo_to_ir_bindings.cc93 llvm_ir::EmitAllocaAtFunctionEntry( in EmitBasePointersForHlos()
Dir_emitter_unnested.cc2269 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitSelectAndScatterFromMlir()
2279 llvm::Value* initialized_flag_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitSelectAndScatterFromMlir()
2358 llvm::Value* select_return_buffer = llvm_ir::EmitAllocaAtFunctionEntry( in EmitSelectAndScatterFromMlir()
2701 llvm::Value* input_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitScatter()
4250 llvm_ir::EmitAllocaAtFunctionEntry(element_type, in EmitPrologueForReduction()
4337 llvm::Value* result_from_other_lane = llvm_ir::EmitAllocaAtFunctionEntry( in EmitFullWarpShuffleDownLoopForReduce()
4499 shared_to_global(llvm_ir::EmitAllocaAtFunctionEntry( in EmitEpilogueForReduction()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.h164 llvm::AllocaInst* EmitAllocaAtFunctionEntry(llvm::Type* type,
Dsort_util.cc121 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/cpu/
Dir_emitter.cc717 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter()
724 llvm::Value* initialized_flag_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter()
1539 accumulator.push_back(llvm_ir::EmitAllocaAtFunctionEntry( in EmitInnerLoopForVectorizedReduction()
3063 llvm::Value* buffer = llvm_ir::EmitAllocaAtFunctionEntry( in EmitThreadLocalBufferPointer()
3248 llvm::Value* parameter_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitThreadLocalCall()
3261 llvm::Value* return_value_buffer = llvm_ir::EmitAllocaAtFunctionEntry( in EmitThreadLocalCall()
Dvector_support_library.cc419 alloca_ = llvm_ir::EmitAllocaAtFunctionEntry(type, "", b_); in LlvmVariable()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2080 llvm::Value* ret_value_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalDynamicUpdateSlice()
2144 llvm::Value* ret_value_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalPad()
2200 llvm_ir::EmitAllocaAtFunctionEntry(primitive_type_llvm, "dot_acc", b_); in EmitElementalDot()
2614 llvm::Value* accum_ptr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalReduceWindow()
2735 llvm::AllocaInst* accumulator_addr = llvm_ir::EmitAllocaAtFunctionEntry( in EmitElementalReduce()
2846 llvm::Value* sum_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitConvolution()