Home
last modified time | relevance | path

Searched refs:FindMatchingPartitionedDimsForGrouping (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.h453 std::optional<std::vector<int64_t>> FindMatchingPartitionedDimsForGrouping(
H A Dspmd_partitioner_util.cc1622 std::optional<std::vector<int64_t>> FindMatchingPartitionedDimsForGrouping( in FindMatchingPartitionedDimsForGrouping() function
H A Ddot_handler.cc2307 if (auto found_dims = FindMatchingPartitionedDimsForGrouping( in GetNonContractingPartitionGroupedShardingForOtherOperand()
2484 if (auto found_dims = FindMatchingPartitionedDimsForGrouping( in GetDotGroupPartitionContractingOutputShardings()