Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4486 llvm::Value* shmem_output_addr = in EmitEpilogueForReduction() local
4489 b_.CreateStore(b_.CreateLoad(current_output), shmem_output_addr); in EmitEpilogueForReduction()
4522 llvm::Value* shmem_output_addr = shared_to_global( in EmitEpilogueForReduction() local
4528 b_.CreateStore(current_output_value, shmem_output_addr); in EmitEpilogueForReduction()