Searched refs:thread_id_x (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3942 llvm::Value* thread_id_x, in GetStartOffsetX() argument
3949 return thread_id_x; in GetStartOffsetX()
3951 return b->CreateMul(thread_id_x, constant(mapping_scheme.GetVectorSize())); in GetStartOffsetX()
3956 return b->CreateMul(thread_id_x, constant(x_num_steps)); in GetStartOffsetX()
4016 mapping_scheme, thread_id_info.thread_id_x, index_ty, &b_); in EmitTile()
4395 llvm::Value* x_loc = thread_id_info.thread_id_x; in EmitEpilogueForReduction()
4484 b_.CreateUDiv(thread_id_info.thread_id_x, constant(kWarpSize)); in EmitEpilogueForReduction()
4504 thread_id_info.thread_id_x, in EmitEpilogueForReduction()
4514 ksl.If("reduction_atomic_update", is_zero(thread_id_info.thread_id_x), in EmitEpilogueForReduction()
4524 thread_id_info.thread_id_x, in EmitEpilogueForReduction()
[all …]
|
| D | ir_emitter_unnested.h | 114 llvm::Value* thread_id_x; member
|