Searched defs:reduction_info (Results 1 – 2 of 2) sorted by relevance
323 ReductionDimensions reduction_info = in SharedMemoryUsage() local
4220 ReductionCodegenInfo* reduction_info) { in EmitPrologueForReduction()4363 const ReductionCodegenInfo& reduction_info) { in GetUntransposedOutputLinearAddress()4382 const ReductionCodegenInfo& reduction_info, in EmitEpilogueForReduction()4602 const ReductionCodegenInfo& reduction_info, int64 x_iter_num) { in EmitTileElementForReduction()5400 ReductionCodegenInfo* reduction_info, const Shape& input_shape) { in EmitIRForReduction()5655 ReductionCodegenInfo reduction_info = in EmitReductionFromOrToContiguousDimensions() local5674 ReductionCodegenInfo reduction_info = in EmitReductionFromOrToContiguousDimensions() local