Searched defs:shard_counts (Results 1 – 1 of 1) sorted by relevance
318 std::vector<int64> shard_counts(dnums.input_spatial_dimensions_size()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local627 std::vector<int64> shard_counts(dnums.input_spatial_dimensions_size()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local