Searched defs:dims_mapping (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 340 const DotConvDimsMapping& dims_mapping, int64 lhs_rank, int64 rhs_rank, in ComputeDimensionIndexMapping()
468 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionBaseCase()
1650 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionDotGroupOnBatch()
1987 const DotConvDimsMapping& dims_mapping, int64 num_partitions, in PartitionDotGroupOnNonContracting()
2081 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionDotGroupOnContracting()
2248 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithFeatureGroupCount()
2271 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithBatchGroupCount()
2288 const DotConvDimsMapping& dims_mapping, const PartitionedHlo& lhs, in LhsIsBestMatchForNonContractingPartitioning()
2433 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionDot()
2800 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionDot()
[all …]
|
| D | convolution_handler.cc | 966 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionConvolution()
|