Searched refs:lhs_shard_sizes (Results 1 – 1 of 1) sorted by relevance
319 std::vector<int64> lhs_shard_sizes(dnums.input_spatial_dimensions_size()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local336 lhs_shard_sizes[i] = lhs_shard_size; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()362 int64 lhs_shard_size = lhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()628 std::vector<int64> lhs_shard_sizes(dnums.input_spatial_dimensions_size()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local645 lhs_shard_sizes[i] = lhs_shard_size; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()660 int64 lhs_shard_size = lhs_shard_sizes[i]; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()