From 6c950a1df0833522b0240703981915ed485f0a55 Mon Sep 17 00:00:00 2001
From: "M. Zeeshan Siddiqui" <mzs@microsoft.com>
Date: Sat, 18 Jul 2020 08:27:02 -0700
Subject: [PATCH] Exclude weight related types/shapes from bert loss. (#4548)

---
 orttraining/orttraining/models/bert/main.cc | 2 --
 1 file changed, 2 deletions(-)

diff --git a/orttraining/orttraining/models/bert/main.cc b/orttraining/orttraining/models/bert/main.cc
index 752211e640..8a3145d664 100644
--- a/orttraining/orttraining/models/bert/main.cc
+++ b/orttraining/orttraining/models/bert/main.cc
@@ -700,14 +700,12 @@ static Status RunPerformanceTest(const BertParameters& params, const Environment
                                             {batch_size, params.max_sequence_length},
                                             {batch_size, params.max_predictions_per_sequence},
                                             {batch_size, params.max_predictions_per_sequence},
-                                            {batch_size, params.max_predictions_per_sequence},
                                             {batch_size}};
   std::vector<onnx::TensorProto_DataType> tensor_types = {onnx::TensorProto_DataType_INT64,
                                                           onnx::TensorProto_DataType_INT64,
                                                           onnx::TensorProto_DataType_INT64,
                                                           onnx::TensorProto_DataType_INT64,
                                                           onnx::TensorProto_DataType_INT64,
-                                                          onnx::TensorProto_DataType_FLOAT,
                                                           onnx::TensorProto_DataType_INT64};
   const size_t num_of_perf_samples = params.num_train_steps * params.batch_size;
   auto random_perf_data = std::make_shared<RandomDataSet>(num_of_perf_samples, tensor_names, tensor_shapes, tensor_types);