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