Home
last modified time | relevance | path

Searched defs:dims_mapping (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc343 const DotConvDimsMapping& dims_mapping, int64_t lhs_rank, int64_t rhs_rank, in ComputeDimensionIndexMapping()
447 const Window& conv_window, const DotConvDimsMapping& dims_mapping, in GetWindowedEinsumConfiguration()
759 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionBaseCase()
2007 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionDotGroupOnBatch()
2354 const DotConvDimsMapping& dims_mapping, int64_t num_partitions, in PartitionDotGroupOnNonContracting()
2459 const DotConvDimsMapping& dims_mapping, const GroupedSharding& lhs_grouped, in GetDotGroupPartitionContractingOutputShardings()
2583 const DotConvDimsMapping& dims_mapping, int64_t num_partitions, in PartitionDotGroupOnContracting()
2797 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithFeatureGroupCount()
2820 const DotConvDimsMapping& dims_mapping, HloInstruction* original_hlo) { in ConvertDimsMappingWithBatchGroupCount()
2842 const DotConvDimsMapping& dims_mapping, const PartitionedHlo& lhs, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
[all …]
H A Dconvolution_handler.cc968 const HloSharding& output_sharding, const DotConvDimsMapping& dims_mapping, in PartitionConvolution()