Home
last modified time | relevance | path

Searched refs:GetReductionKindAndContiguousComponents (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emission_utils.h93 ReductionDimensions GetReductionKindAndContiguousComponents(
95 ReductionDimensions GetReductionKindAndContiguousComponents(
H A Dir_emission_utils.cc293 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents() function
299 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents() function
H A Dtree_reduction_rewriter.cc79 GetReductionKindAndContiguousComponents(*hlo); in RewriteReduction()
H A Dgpu_fusible.cc300 GetReductionKindAndContiguousComponents(instr); in SharedMemoryUsageNoCache()
H A Dir_emitter_unnested.cc4573 GetReductionKindAndContiguousComponents(first_reduce); in ComputeReductionCodegenInfo()