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

Fix a crash in deep_cpu_gru_op_test.cc (#2028)

Browse source
This commit is contained in:
Changming Sun 2019-10-08 10:03:07 -07:00 committed by GitHub
parent 71b389322e
commit 3053af812c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/providers/cpu/rnn/deep_cpu_gru_op_test.cc
View file

@ -12,6 +12,8 @@ using namespace std;
namespace onnxruntime {
namespace test {
static const std::vector<string> default_activations = {"sigmoid", "tanh"};
static void RunGruTest(const std::vector<float>& X_data,
const std::vector<float>& W_data,
const std::vector<float>& R_data,
@ -29,7 +31,7 @@ static void RunGruTest(const std::vector<float>& X_data,
bool output_sequence = true,
bool linear_before_reset = false,
// copy the following vectors as we may modify them
std::vector<string> activations = {"sigmoid", "tanh"},
std::vector<string> activations = default_activations,
std::vector<float> activation_alphas = {},
std::vector<float> activation_betas = {}) {
OpTester test("GRU");