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.cc79 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b_); in EmitIndexAndSetExitBasicBlock()
89 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b_); in EmitIndexAndSetExitBasicBlock()
Delemental_ir_emitter.cc308 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b()), in EmitThreadId()
311 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b()), in EmitThreadId()
314 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockDimx, {}, {}, b()), in EmitThreadId()
Dtarget_util.h78 llvm::CallInst* EmitCallToTargetIntrinsic(
Dtarget_util.cc223 llvm::CallInst* EmitCallToTargetIntrinsic( in EmitCallToTargetIntrinsic() function
Dir_emission_utils.cc586 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b)); in IsBlock0Thread0()
590 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b)); in IsBlock0Thread0()
Dir_emitter_unnested.cc4565 return gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBlockIdx, {}, in EmitBlockId()
4673 llvm::CallInst* thread_id_raw = gpu::EmitCallToTargetIntrinsic( in EmitThreadId()
4789 return EmitCallToTargetIntrinsic(TargetIntrinsicID::kBarrierId, {}, {}, &b_); in EmitSyncThreads()
5666 llvm::CallInst* raw_block_id_y = gpu::EmitCallToTargetIntrinsic( in EmitReductionFromOrToContiguousDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc153 llvm::Value* thread_id = gpu::EmitCallToTargetIntrinsic( in EmitTiledCompareLoop()
205 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, b); in EmitTiledCompareLoop()
266 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, in EmitTiledCompareLoop()