Searched refs:desired_operand_sharding (Results 1 – 1 of 1) sorted by relevance
1956 auto desired_operand_sharding = in HandleTranspose() local1960 .Reshard(desired_operand_sharding) in HandleTranspose()1978 auto desired_operand_sharding = hlo_sharding_util::ReshapeSharding( in HandleReshape() local1980 if (desired_operand_sharding.has_value()) { in HandleReshape()1981 auto operand_hlo = operand.Reshard(*desired_operand_sharding).hlo(); in HandleReshape()