Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1917 absl::Span<const DotConvDimsMapping::DimsMapping> matching_partitioned_dims, in GetNonContractingPartitionGroupedShardingForOtherOperand() argument
1923 for (const auto& dim : matching_partitioned_dims) { in GetNonContractingPartitionGroupedShardingForOtherOperand()