Searched refs:TargetIntrinsicID (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | target_util.cc | 47 struct TargetIntrinsics GetIntrinsic(TargetIntrinsicID intrin) { in GetIntrinsic()
49 case TargetIntrinsicID::kThreadIdx: { in GetIntrinsic()
53 case TargetIntrinsicID::kThreadIdy: { in GetIntrinsic()
57 case TargetIntrinsicID::kThreadIdz: { in GetIntrinsic()
61 case TargetIntrinsicID::kBlockIdx: { in GetIntrinsic()
65 case TargetIntrinsicID::kBlockIdy: { in GetIntrinsic()
69 case TargetIntrinsicID::kBlockIdz: { in GetIntrinsic()
73 case TargetIntrinsicID::kBarrierId: { in GetIntrinsic()
77 case TargetIntrinsicID::kBlockDimx: { in GetIntrinsic()
85 case TargetIntrinsicID::kBlockDimy: { in GetIntrinsic()
[all …]
|
| D | target_util.h | 34 enum class TargetIntrinsicID { enum
79 TargetIntrinsicID intrinsic_id, absl::Span<llvm::Value* const> operands,
|
| D | parallel_loop_emitter.cc | 62 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b_); in EmitLinearBaseAndThreadIdx()
70 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b_); in EmitLinearBaseAndThreadIdx()
97 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdy, {}, {}, b_); in EmitLinearBaseAndThreadIdx()
|
| D | elemental_ir_emitter.cc | 324 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b()), in EmitThreadId()
327 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b()), in EmitThreadId()
330 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockDimx, {}, {}, b()), in EmitThreadId()
|
| D | ir_emission_utils.cc | 450 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b)); in IsBlock0Thread0()
454 EmitCallToTargetIntrinsic(TargetIntrinsicID::kBlockIdx, {}, {}, b)); in IsBlock0Thread0()
|
| D | ir_emitter_unnested.cc | 3789 gpu::TargetIntrinsicID::kBlockIdx, {}, {}, &b_); in EmitBlockId()
4060 gpu::TargetIntrinsicID::kThreadIdx, {}, {}, &b_); in EmitThreadId()
4198 return EmitCallToTargetIntrinsic(TargetIntrinsicID::kBarrierId, {}, {}, &b_); in EmitSyncThreads()
4985 gpu::TargetIntrinsicID::kBlockIdy, {}, {}, &b_); in EmitUnnestedReduction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 164 gpu::TargetIntrinsicID::kThreadIdx, {}, {}, b); in EmitTiledCompareLoop()
216 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, b); in EmitTiledCompareLoop()
286 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, in EmitTiledCompareLoop()
|