Home
last modified time | relevance | path

Searched defs:dims_to_keep (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emission_utils.cc156 DimensionVector dims_to_keep; in GetReductionKindAndContiguousComponentsImpl() local
224 DimensionVector dims_to_keep; in IsReductionFromOrToContiguousDimensionsImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.cc1311 const HloSharding& sharding, absl::Span<const int64_t> dims_to_keep) { in PartiallyReplicateTiledShardingOnAllDimsExcept()