Searched defs:mapping_scheme (Results 1 – 2 of 2) sorted by relevance
169 explicit ReductionCodegenInfo(KernelMappingScheme mapping_scheme, in ReductionCodegenInfo()
3489 static llvm::Value* GetStartOffsetX(const KernelMappingScheme& mapping_scheme, in GetStartOffsetX()3549 const KernelMappingScheme& mapping_scheme, in EmitTile()3660 const KernelMappingScheme& mapping_scheme, llvm::Value* y_loc, in EmitTileElementForFusion()3787 auto& mapping_scheme = reduction_info->GetKernelMappingScheme(); in EmitPrologueForReduction() local3888 const KernelMappingScheme& mapping_scheme = in EmitEpilogueForReduction() local4036 const KernelMappingScheme& mapping_scheme = in EmitEpilogueForRowReduction() local4103 const KernelMappingScheme& mapping_scheme = in EmitEpilogueForColumnReduction() local4246 const KernelMappingScheme& mapping_scheme, llvm::Type* index_ty, in EmitTilingKernel()4373 const KernelMappingScheme& mapping_scheme, in EmitHlo021Tile()4691 KernelMappingScheme mapping_scheme(*reduced_dims_021, in CheckAndEmitHloWithTile021() local[all …]