Searched refs:lhs_matching (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1881 bool lhs_matching, const HloSharding& matching_sharding, in GetNonContractingPartitionGroupedShardingForMatchedOperand() argument
1891 int64 md = lhs_matching ? dim.lhs : dim.rhs; in GetNonContractingPartitionGroupedShardingForMatchedOperand()
1912 bool lhs_matching, const Shape& output_base_shape, const Shape& other_shape, in GetNonContractingPartitionGroupedShardingForOtherOperand() argument
1951 other_group_dims.push_back(lhs_matching ? dim.rhs : dim.lhs); in GetNonContractingPartitionGroupedShardingForOtherOperand()
1955 other_group_dims.push_back(lhs_matching ? dim.rhs : dim.lhs); in GetNonContractingPartitionGroupedShardingForOtherOperand()
1980 bool lhs_matching, PartitionedHlo matching, PartitionedHlo other, in PartitionDotGroupOnNonContracting() argument
2012 lhs_matching, matching.sharding(), output_sharding, in PartitionDotGroupOnNonContracting()
2020 lhs_matching, output_base_shape, other.hlo()->shape(), in PartitionDotGroupOnNonContracting()
2025 lhs_matching ? dims_mapping.rhs_non_contracting_dims in PartitionDotGroupOnNonContracting()
2065 PartitionDot(lhs_matching ? matching_p : other_p, in PartitionDotGroupOnNonContracting()
[all …]
|