Searched refs:kThreadIdx (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | target_util.h | 35 kThreadIdx = 0, enumerator
|
| D | parallel_loop_emitter.cc | 89 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b_); in EmitIndexAndSetExitBasicBlock()
|
| D | target_util.cc | 48 case TargetIntrinsicID::kThreadIdx: { in GetIntrinsic()
|
| D | elemental_ir_emitter.cc | 311 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b()), in EmitThreadId()
|
| D | ir_emission_utils.cc | 586 EmitCallToTargetIntrinsic(TargetIntrinsicID::kThreadIdx, {}, {}, b)); in IsBlock0Thread0()
|
| D | ir_emitter_unnested.cc | 4674 gpu::TargetIntrinsicID::kThreadIdx, {}, {}, &b_); in EmitThreadId()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 154 gpu::TargetIntrinsicID::kThreadIdx, {}, {}, b); in EmitTiledCompareLoop()
|