Unverified Commit d8199edc authored by Jayaram Bobba's avatar Jayaram Bobba Committed by GitHub

Merge branch 'master' into jmenon/mkldnn-compile

parents fd869f1e 292ab46f
...@@ -283,7 +283,7 @@ namespace ngraph ...@@ -283,7 +283,7 @@ namespace ngraph
// define weights // define weights
writer << "std::vector<" << args[0].get_element_type().c_type_string() writer << "std::vector<" << args[0].get_element_type().c_type_string()
<< ">bn_weights(2);\n"; << ">bn_weights(2*" << input_shape[1] << ");\n";
auto weights_shape = Shape{2, input_shape[1]}; auto weights_shape = Shape{2, input_shape[1]};
// push gamma and beta // push gamma and beta
......
...@@ -155,13 +155,15 @@ TEST(benchmark, mxnet_seq2seq_backward) ...@@ -155,13 +155,15 @@ TEST(benchmark, mxnet_seq2seq_backward)
TEST(benchmark, mxnet_sockeye_seq2seq_forward) TEST(benchmark, mxnet_sockeye_seq2seq_forward)
{ {
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/Sockeye_Seq2Seq_forward.json"); const string json_path =
file_util::path_join(SERIALIZED_ZOO, "mxnet/Sockeye_Seq2Seq_forward.json");
run_benchmark(json_path, "CPU", 10); run_benchmark(json_path, "CPU", 10);
} }
TEST(benchmark, mxnet_sockeye_seq2seq_backward) TEST(benchmark, mxnet_sockeye_seq2seq_backward)
{ {
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/Sockeye_Seq2Seq_backward.json"); const string json_path =
file_util::path_join(SERIALIZED_ZOO, "mxnet/Sockeye_Seq2Seq_backward.json");
run_benchmark(json_path, "CPU", 10); run_benchmark(json_path, "CPU", 10);
} }
......
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