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.h226 ReductionDimensions GetReductionKindAndContiguousComponents(
228 ReductionDimensions GetReductionKindAndContiguousComponents(
Dtree_reduction_rewriter.cc67 GetReductionKindAndContiguousComponents(*hlo); in RewriteReduction()
Dir_emission_utils.cc299 GetReductionKindAndContiguousComponents(reduce)); in IsReductionFromOrToContiguousDimensions()
482 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents() function
488 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents() function
Dgpu_fusible.cc323 GetReductionKindAndContiguousComponents(instr); in SharedMemoryUsage()
Dir_emitter_unnested.cc4828 GetReductionKindAndContiguousComponents(first_reduce); in ComputeReductionCodegenInfo()