Searched refs:partitioned_dims (Results 1 – 2 of 2) sorted by relevance
285 partitioned_dims = (1,) * dims_to_add + self._partitioned_dim_sizes286 return RaggedTensorDynamicShape(partitioned_dims, self._inner_dim_sizes)
1883 absl::Span<const DotConvDimsMapping::DimsMapping> partitioned_dims) { in GetNonContractingPartitionGroupedShardingForMatchedOperand() argument1890 for (const auto& dim : partitioned_dims) { in GetNonContractingPartitionGroupedShardingForMatchedOperand()