Searched refs:thread_id_y (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 117 llvm::Value* thread_id_y; member
|
| D | ir_emitter_unnested.cc | 4057 ceil_of_ratio(b_.CreateSub(tile_height, thread_id_info.thread_id_y), in EmitTile()
4061 thread_id_info.thread_id_y, b_.CreateMul(y_indvar, num_threads_y)); in EmitTile()
4396 llvm::Value* y_loc = thread_id_info.thread_id_y; in EmitEpilogueForReduction()
4525 thread_id_info.thread_id_y}), in EmitEpilogueForReduction()
4536 {b_.getInt32(0), constant(j), thread_id_info.thread_id_y, in EmitEpilogueForReduction()
4548 GetStartOffsetX(mapping_scheme, thread_id_info.thread_id_y, in EmitEpilogueForReduction()
|