Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc96 auto aligned_rhs_sharding = in PartitionConvolutionWithBatchGroupCount() local
187 auto aligned_rhs_sharding = in PartitionConvolutionWithFeatureGroupCount() local
253 auto aligned_rhs_sharding = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
562 auto aligned_rhs_sharding = hlo_sharding_util::ReverseSharding( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local