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.h192 ReductionDimensions GetReductionKindAndContiguousComponents(
194 ReductionDimensions GetReductionKindAndContiguousComponents(
Dir_emission_utils.cc285 GetReductionKindAndContiguousComponents(reduce); in IsReductionFromOrToContiguousDimensions()
397 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents() function
403 ReductionDimensions GetReductionKindAndContiguousComponents( in GetReductionKindAndContiguousComponents() function
Dtree_reduction_rewriter.cc73 GetReductionKindAndContiguousComponents(*hlo); in RewriteReduction()
Dgpu_fusible.cc324 GetReductionKindAndContiguousComponents(instr); in SharedMemoryUsage()
Dir_emitter_unnested.cc5298 GetReductionKindAndContiguousComponents(first_reduce); in ComputeReductionCodegenInfo()