Home
last modified time | relevance | path

Searched refs:EmitLinearBaseAndThreadIdx (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dparallel_loop_emitter.cc59 ParallelLoopEmitter::EmitLinearBaseAndThreadIdx(llvm::Type* index_type, in EmitLinearBaseAndThreadIdx() function in xla::gpu::ParallelLoopEmitter
153 EmitLinearBaseAndThreadIdx(index_type, base_index); in EmitLogicalIndexAndSetExitBasicBlock()
242 EmitLinearBaseAndThreadIdx(index_type, base_index); in EmitIndexAndSetExitBasicBlock()
H A Dparallel_loop_emitter.h76 LinearBaseAndThreadIdx EmitLinearBaseAndThreadIdx(llvm::Type* index_type,