Home
last modified time | relevance | path

Searched defs:num_partial_results (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_mapping_scheme.h170 int num_partial_results, bool is_row_reduction, in ReductionCodegenInfo()
Dir_emitter_unnested.cc3761 int num_partial_results = reduction_info->GetNumPartialResults(); in EmitPrologueForReduction() local
4190 int num_partial_results = reduction_info.GetNumPartialResults(); in EmitTileElementForReduction() local
4878 int num_partial_results = 1; in ComputeReductionCodegenInfo() local