Searched refs:GatherParallelDimSharding (Results 1 – 2 of 2) sorted by relevance
34 struct GatherParallelDimSharding { struct415 absl::optional<GatherParallelDimSharding>
1831 absl::optional<GatherParallelDimSharding>1854 return GatherParallelDimSharding{ in GatherOperandsShardedAcrossParallelDims()1861 return GatherParallelDimSharding{ in GatherOperandsShardedAcrossParallelDims()1933 return GatherParallelDimSharding{new_index_shard, new_operand_shard}; in GatherOperandsShardedAcrossParallelDims()