Searched defs:mapping_scheme (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | kernel_mapping_scheme.h | 171 explicit ReductionCodegenInfo(KernelMappingScheme mapping_scheme, in ReductionCodegenInfo()
|
| D | ir_emitter_unnested.cc | 3941 static llvm::Value* GetStartOffsetX(const KernelMappingScheme& mapping_scheme, in GetStartOffsetX()
4001 const KernelMappingScheme& mapping_scheme, in EmitTile()
4099 const KernelMappingScheme& mapping_scheme, llvm::Value* y_loc, in EmitTileElementForCopy()
4148 const KernelMappingScheme& mapping_scheme, llvm::Value* y_loc, in EmitTileElementForFusion()
4286 auto& mapping_scheme = reduction_info->GetKernelMappingScheme(); in EmitPrologueForReduction() local
4384 const KernelMappingScheme& mapping_scheme = in EmitEpilogueForReduction() local
4695 const KernelMappingScheme& mapping_scheme, llvm::Type* index_ty, in EmitTilingKernel()
4826 KernelMappingScheme mapping_scheme(reduced_output_dims, in EmitHlo021Tile() local
5386 KernelMappingScheme mapping_scheme( in ComputeReductionCodegenInfo() local
5426 const KernelMappingScheme& mapping_scheme = in EmitIRForReduction() local
[all …]
|