Skip to content

GitLab

Commit 690dd868 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 544811c3
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/test_runner.py
View file @ 690dd868
...@@ -95,6 +95,7 @@ def model_parameter_names(model_file_name): ...@@ -95,6 +95,7 @@ def model_parameter_names(model_file_name):
return param_names return param_names
def model_output_names(model_file_name): def model_output_names(model_file_name):
with open(model_file_name, 'rb') as pfile: with open(model_file_name, 'rb') as pfile:
data_str = pfile.read() data_str = pfile.read()
...@@ -104,6 +105,7 @@ def model_output_names(model_file_name): ...@@ -104,6 +105,7 @@ def model_output_names(model_file_name):
return output_names return output_names
def get_input_shapes(sample_case, param_names): def get_input_shapes(sample_case, param_names):
param_shape_map = {} param_shape_map = {}
for i in range(len(param_names)): for i in range(len(param_names)):
...@@ -147,8 +149,6 @@ def check_correctness(gold_outputs, outputs, rtol=1e-3, atol=1e-3): ...@@ -147,8 +149,6 @@ def check_correctness(gold_outputs, outputs, rtol=1e-3, atol=1e-3):
out_num = len(gold_outputs) out_num = len(gold_outputs)
ret = True ret = True
for i in range(out_num): for i in range(out_num):
# print("Expected value: \n{}".format(gold_outputs[i]))
# print("Actual value: \n{}".format(outputs[i]))
if not np.allclose(gold_outputs[i], outputs[i], rtol, atol): if not np.allclose(gold_outputs[i], outputs[i], rtol, atol):
print("Output {} is incorrect ...".format(i)) print("Output {} is incorrect ...".format(i))
print("Expected value: \n{}".format(gold_outputs[i])) print("Expected value: \n{}".format(gold_outputs[i]))
......
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