Searched refs:EmitThreadIdInfo (Results 1 – 2 of 2) sorted by relevance
709 ThreadIdInfo EmitThreadIdInfo(int64 threads_per_block, llvm::Type* index_ty,
4391 EmitThreadIdInfo(mapping_scheme.GetThreadsPerBlock(), index_ty, in EmitEpilogueForReduction()4680 IrEmitterUnnested::ThreadIdInfo IrEmitterUnnested::EmitThreadIdInfo( in EmitThreadIdInfo() function in xla::gpu::IrEmitterUnnested4707 EmitThreadIdInfo(mapping_scheme.GetThreadsPerBlock(), index_ty, in EmitTilingKernel()