Searched refs:ReductionDimensions (Results 1 – 5 of 5) sorted by relevance
174 struct ReductionDimensions { struct192 ReductionDimensions GetReductionKindAndContiguousComponents(194 ReductionDimensions GetReductionKindAndContiguousComponents(202 const ReductionDimensions& reduction_dimensions,
134 const ReductionDimensions& reduction_dimensions, in GetReductionTiling()216 static ReductionDimensions GetReductionKindAndContiguousComponentsImpl( in GetReductionKindAndContiguousComponentsImpl()284 ReductionDimensions reduction_dimensions = in IsReductionFromOrToContiguousDimensions()356 ReductionDimensions reduction_dimensions = in IsReductionFromOrToContiguousDimensions()397 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents()403 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents()
72 ReductionDimensions reduction_dimensions = in RewriteReduction()201 HloInstruction *hlo, const ReductionDimensions &reduction_dimensions, in RewriteBatchDimensionLargerThanTile()
323 ReductionDimensions reduction_info = in SharedMemoryUsage()
5297 ReductionDimensions reduction_dimensions = in ComputeReductionCodegenInfo()