Home
last modified time | relevance | path

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

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