Searched refs:TargetIntrinsicID (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | target_util.cc | 46 struct TargetIntrinsics GetIntrinsic(TargetIntrinsicID intrin) { in GetIntrinsic()
48 case TargetIntrinsicID::kThreadIdx: { in GetIntrinsic()
52 case TargetIntrinsicID::kThreadIdy: { in GetIntrinsic()
56 case TargetIntrinsicID::kThreadIdz: { in GetIntrinsic()
60 case TargetIntrinsicID::kBlockIdx: { in GetIntrinsic()
64 case TargetIntrinsicID::kBlockIdy: { in GetIntrinsic()
68 case TargetIntrinsicID::kBlockIdz: { in GetIntrinsic()
72 case TargetIntrinsicID::kBarrierId: { in GetIntrinsic()
76 case TargetIntrinsicID::kBlockDimx: { in GetIntrinsic()
84 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 | 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 | 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()
4674 gpu::TargetIntrinsicID::kThreadIdx, {}, {}, &b_); in EmitThreadId()
4789 return EmitCallToTargetIntrinsic(TargetIntrinsicID::kBarrierId, {}, {}, &b_); in EmitSyncThreads()
5667 gpu::TargetIntrinsicID::kBlockIdy, {}, {}, &b_); in EmitReductionFromOrToContiguousDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 154 gpu::TargetIntrinsicID::kThreadIdx, {}, {}, b); in EmitTiledCompareLoop()
205 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, b); in EmitTiledCompareLoop()
266 gpu::EmitCallToTargetIntrinsic(gpu::TargetIntrinsicID::kBarrierId, {}, {}, in EmitTiledCompareLoop()
|