Home
last modified time | relevance | path

Searched refs:ReverseSharding (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h96 HloSharding ReverseSharding(const HloSharding& sharding,
Dsharding_propagation.cc852 hlo_sharding_util::ReverseSharding( in InferShardingFromOperands()
1350 return hlo_sharding_util::ReverseSharding(user.sharding(), in GetShardingFromUser()
Dhlo_sharding_util.cc412 HloSharding ReverseSharding(const HloSharding& sharding, in ReverseSharding() function
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc562 auto aligned_rhs_sharding = hlo_sharding_util::ReverseSharding( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
566 hlo_sharding_util::ReverseSharding(rhs.sharding(), reversed_rhs_dims), in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
Dspmd_partitioner.cc2823 .Reshard(hlo_sharding_util::ReverseSharding( in HandleReverse()