Home
last modified time | relevance | path

Searched refs:reduction_info (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h594 const ReductionCodegenInfo& reduction_info, int64 x_iter_num);
605 ReductionCodegenInfo* reduction_info);
614 const ReductionCodegenInfo& reduction_info,
624 ReductionCodegenInfo* reduction_info,
Dir_emitter_unnested.cc4220 ReductionCodegenInfo* reduction_info) { in EmitPrologueForReduction() argument
4246 reduction_info->GetMutableReductionInputAddresses(); in EmitPrologueForReduction()
4254 int num_partial_results = reduction_info->GetNumPartialResults(); in EmitPrologueForReduction()
4256 reduction_info->GetMutablePartialResultAddresses(); in EmitPrologueForReduction()
4284 reduction_info->GetMutableInitialValues()->push_back(init_ir_value); in EmitPrologueForReduction()
4286 auto& mapping_scheme = reduction_info->GetKernelMappingScheme(); in EmitPrologueForReduction()
4291 if (reduction_info->IsRowReduction()) { in EmitPrologueForReduction()
4313 reduction_info->GetMutableSharedCache()->push_back(shared_cache_per_reduce); in EmitPrologueForReduction()
4363 const ReductionCodegenInfo& reduction_info) { in GetUntransposedOutputLinearAddress() argument
4365 reduction_info.GetKernelMappingScheme(); in GetUntransposedOutputLinearAddress()
[all …]
Dgpu_fusible.cc323 ReductionDimensions reduction_info = in SharedMemoryUsage() local
327 if (reduction_info.is_row_reduction) { in SharedMemoryUsage()