Skip to content
Snippets Groups Projects
Commit c2ac2947 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Transform] mark threshold node domain as finn

parent b0941313
No related branches found
No related tags found
No related merge requests found
...@@ -15,6 +15,7 @@ def convert_sign_to_thres(model): ...@@ -15,6 +15,7 @@ def convert_sign_to_thres(model):
thres_param_name = model.make_new_valueinfo_name() thres_param_name = model.make_new_valueinfo_name()
thres_param = np.asarray([[0]], dtype=np.float32) thres_param = np.asarray([[0]], dtype=np.float32)
n.input.append(thres_param_name) n.input.append(thres_param_name)
n.domain = "finn"
model.set_initializer(thres_param_name, thres_param) model.set_initializer(thres_param_name, thres_param)
# mark output tensor as bipolar # mark output tensor as bipolar
model.set_tensor_datatype(n.output[0], DataType.BIPOLAR) model.set_tensor_datatype(n.output[0], DataType.BIPOLAR)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment