Commit 5ad1de22 authored by Robert Kimball's avatar Robert Kimball Committed by Matthew Brookhart

Add mxnet seq2seq serialized model for benchmarking (#385)

* add mxnet seq2seq forward and backward

* add benchmarks for seq2seq forward and backward
parent b6c98de1
......@@ -127,34 +127,28 @@ TEST(benchmark, mxnet_10_bucket_lstm)
run_benchmark(json_path, "CPU", 10);
}
TEST(benchmark, mxnet_10_bucket_lstm_int)
{
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/10_bucket_LSTM.json");
run_benchmark(json_path, "INTERPRETER", 10);
}
TEST(benchmark, mxnet_lstm_backward)
{
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_backward.json");
run_benchmark(json_path, "CPU", 10);
}
TEST(benchmark, mxnet_lstm_backward_int)
TEST(benchmark, mxnet_lstm_forward)
{
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_backward.json");
run_benchmark(json_path, "INTERPRETER", 1);
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_forward.json");
run_benchmark(json_path, "CPU", 10);
}
TEST(benchmark, mxnet_lstm_forward)
TEST(benchmark, mxnet_seq2seq_forward)
{
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_forward.json");
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/Seq2Seq_forward.json");
run_benchmark(json_path, "CPU", 10);
}
TEST(benchmark, mxnet_lstm_forward_int)
TEST(benchmark, mxnet_seq2seq_backward)
{
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_forward.json");
run_benchmark(json_path, "INTERPRETER", 10);
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/Seq2Seq_backward.json");
run_benchmark(json_path, "CPU", 10);
}
//
......
This source diff could not be displayed because it is too large. You can view the blob instead.
This source diff could not be displayed because it is too large. You can view the blob instead.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment