Searched refs:partial_result_address (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 4257 llvm::AllocaInst* partial_result_address = in EmitPrologueForReduction() local
4261 partial_result_addresses->push_back(partial_result_address); in EmitPrologueForReduction()
4282 InBoundsGEP(partial_result_address, {b_.getInt32(i)})); in EmitPrologueForReduction()
4332 llvm::Value* partial_result_address, int threads_per_block) { in EmitFullWarpShuffleDownLoopForReduce() argument
4348 Load(convert_pointer_for_shuffle(partial_result_address), in EmitFullWarpShuffleDownLoopForReduce()
4353 *reducer, {partial_result_address, result_from_other_lane}, in EmitFullWarpShuffleDownLoopForReduce()
4354 partial_result_address)); in EmitFullWarpShuffleDownLoopForReduce()
4655 llvm::Value* partial_result_address = InBoundsGEP( in EmitTileElementForReduction() local
4658 *reducers[i], {partial_result_address, input_address}, in EmitTileElementForReduction()
4659 partial_result_address)); in EmitTileElementForReduction()
|
| D | ir_emitter_unnested.h | 638 llvm::Value* partial_result_address,
|