Commit 69838eb5 authored by Sebastian N.'s avatar Sebastian N.

Updated tests

parent 91616996
Pipeline #205608 failed with stages
in 27 seconds
......@@ -44,7 +44,7 @@ public:
MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
size = 1;
for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
assert(size == out_predictions_.size());
assert(out_predictions_.size() == 1 || size == out_predictions_.size());
MXPredGetOutput(handle, output_index, &(out_predictions_[0]), out_predictions_.size());
}
......
......@@ -44,7 +44,7 @@ public:
MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
size = 1;
for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
assert(size == out_qvalues_.size());
assert(out_qvalues_.size() == 1 || size == out_qvalues_.size());
MXPredGetOutput(handle, output_index, &(out_qvalues_[0]), out_qvalues_.size());
}
......
......@@ -44,7 +44,7 @@ public:
MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
size = 1;
for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
assert(size == out_action_.size());
assert(out_action_.size() == 1 || size == out_action_.size());
MXPredGetOutput(handle, output_index, &(out_action_[0]), out_action_.size());
}
......
......@@ -44,7 +44,7 @@ public:
MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
size = 1;
for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
assert(size == out_qvalues_.size());
assert(out_qvalues_.size() == 1 || size == out_qvalues_.size());
MXPredGetOutput(handle, output_index, &(out_qvalues_[0]), out_qvalues_.size());
}
......
......@@ -44,7 +44,7 @@ public:
MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
size = 1;
for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
assert(size == out_commands_.size());
assert(out_commands_.size() == 1 || size == out_commands_.size());
MXPredGetOutput(handle, output_index, &(out_commands_[0]), out_commands_.size());
}
......
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