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) ...@@ -127,34 +127,28 @@ TEST(benchmark, mxnet_10_bucket_lstm)
run_benchmark(json_path, "CPU", 10); 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) TEST(benchmark, mxnet_lstm_backward)
{ {
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_backward.json"); const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_backward.json");
run_benchmark(json_path, "CPU", 10); 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"); const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/LSTM_forward.json");
run_benchmark(json_path, "INTERPRETER", 1); 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); 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"); const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/Seq2Seq_backward.json");
run_benchmark(json_path, "INTERPRETER", 10); run_benchmark(json_path, "CPU", 10);
} }
// //
......
This diff is collapsed.
This diff is collapsed.
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