Searched refs:ThreadIdInfo (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 109 struct ThreadIdInfo { struct
142 const ThreadIdInfo& thread_id_info, const llvm_ir::IrArray::Index& index, argument
551 KernelSupportLibrary* ksl, const ThreadIdInfo& thread_id_info,
709 ThreadIdInfo EmitThreadIdInfo(int64 threads_per_block, llvm::Type* index_ty,
|
| D | ir_emitter_unnested.cc | 4003 KernelSupportLibrary* ksl, const ThreadIdInfo& thread_id_info, in EmitTile()
4390 IrEmitterUnnested::ThreadIdInfo thread_id_info = in EmitEpilogueForReduction()
4680 IrEmitterUnnested::ThreadIdInfo IrEmitterUnnested::EmitThreadIdInfo( in EmitThreadIdInfo()
4706 IrEmitterUnnested::ThreadIdInfo thread_id_info = in EmitTilingKernel()
4898 [&](const ThreadIdInfo& thread_id_info, const IrArray::Index& index, in EmitHlo021Tile()
5447 [&](const ThreadIdInfo& thread_id_info, const IrArray::Index& index, in EmitIRForReduction()
|