Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_mapping_scheme.h182 const KernelMappingScheme& GetKernelMappingScheme() const { in GetKernelMappingScheme() function
Dir_emitter_unnested.cc4286 auto& mapping_scheme = reduction_info->GetKernelMappingScheme(); in EmitPrologueForReduction()
4365 reduction_info.GetKernelMappingScheme(); in GetUntransposedOutputLinearAddress()
4385 reduction_info.GetKernelMappingScheme(); in EmitEpilogueForReduction()
4634 reduction_info.GetKernelMappingScheme()); in EmitTileElementForReduction()
5427 reduction_info->GetKernelMappingScheme(); in EmitIRForReduction()
5450 EmitTile(reduction_info->GetKernelMappingScheme(), index, loop_name, in EmitIRForReduction()
5677 reduction_info.GetKernelMappingScheme(); in EmitReductionFromOrToContiguousDimensions()