Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_canonicalize_reduction.cc149 SmallVector<int64_t, 4> dims_to_keep; in runOnFunction() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc217 DimensionVector dims_to_keep; in GetReductionKindAndContiguousComponentsImpl() local
280 std::vector<int64> dims_to_keep; in IsReductionFromOrToContiguousDimensions() local
435 std::vector<int64> dims_to_keep; in IsReductionFromOrToContiguousDimensions() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc1230 const HloSharding& sharding, absl::Span<const int64> dims_to_keep) { in PartiallyReplicateTiledShardingOnAllDimsExcept()