saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge remote-tracking branch 'upstream/master' into DmlDev

Browse source
This commit is contained in:
ISS Build Account 2020-07-19 11:07:01 +00:00
commit a15506ee84
1 changed files with 0 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/models/bert/main.cc
View file

@ -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);