Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc62 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b_); in EmitLinearBaseAndThreadIdx()
70 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b_); in EmitLinearBaseAndThreadIdx()
97 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdy, {}, {}, b_); in EmitLinearBaseAndThreadIdx()
Delemental_ir_emitter.cc324 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b()), in EmitThreadId()
327 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b()), in EmitThreadId()
330 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockDimx, {}, {}, b()), in EmitThreadId()
Dtarget_util.h78 llvm::CallInst* EmitCallToTargetIntrinsic(
Dtarget_util.cc224 llvm::CallInst* EmitCallToTargetIntrinsic( in EmitCallToTargetIntrinsic() function
Dir_emission_utils.cc450 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b)); in IsBlock0Thread0()
454 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b)); in IsBlock0Thread0()
Dir_emitter_unnested.cc3788 llvm::Value* block_id = gpu::EmitCallToTargetIntrinsic( in EmitBlockId()
4059 llvm::CallInst* thread_id_raw = gpu::EmitCallToTargetIntrinsic( in EmitThreadId()
4198 return EmitCallToTargetIntrinsic(TargetIntrinsicID::kBarrierId, {}, {}, &b_); in EmitSyncThreads()
4984 llvm::CallInst* raw_block_id_y = gpu::EmitCallToTargetIntrinsic( in EmitUnnestedReduction()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc163 llvm::Value* thread_id = gpu::EmitCallToTargetIntrinsic( in EmitTiledCompareLoop()
216 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, b); in EmitTiledCompareLoop()
286 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, in EmitTiledCompareLoop()