message: [XLA] Update cost model to account for batch group counts.
diff: @@ -524,7 +524,8 @@ Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) {
  }
  const int64 fma_count = (input_feature / convolution->feature_group_count()) *
  - output_feature * batch *
  + output_feature *
  + (batch / convolution->batch_group_count()) *
  Product(valid_position_counts);
  current_properties_[kFlopsKey] = fma_count * kFmaFlops;
  return Status::OK();
  