Searched refs:EmitCallToTargetIntrinsic (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | parallel_loop_emitter.cc | 79 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b_); in EmitIndexAndSetExitBasicBlock() 89 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b_); in EmitIndexAndSetExitBasicBlock()
|
D | elemental_ir_emitter.cc | 308 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b()), in EmitThreadId() 311 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b()), in EmitThreadId() 314 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockDimx, {}, {}, b()), in EmitThreadId()
|
D | target_util.h | 78 llvm::CallInst* EmitCallToTargetIntrinsic(
|
D | target_util.cc | 223 llvm::CallInst* EmitCallToTargetIntrinsic( in EmitCallToTargetIntrinsic() function
|
D | ir_emission_utils.cc | 586 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b)); in IsBlock0Thread0() 590 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b)); in IsBlock0Thread0()
|
D | ir_emitter_unnested.cc | 4565 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/ |
D | sort_util.cc | 153 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()
|