Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h168 llvm::Value* EmitFullWarpShuffleDown(llvm::Value* value, llvm::Value* offset,
Dir_emission_utils.cc209 llvm::Value* EmitFullWarpShuffleDown(llvm::Value* value, llvm::Value* offset, in EmitFullWarpShuffleDown() function
Dir_emitter_unnested.cc2769 Store(EmitFullWarpShuffleDown(partial_result, b_.getInt32(distance), &b_), in EmitFullWarpShuffleDownLoopForAllReduces()