We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e5452f8 commit d43b22aCopy full SHA for d43b22a
build_rknn.py
@@ -72,7 +72,7 @@ def ConvertModel(model_path='ViT-B-32__openai/textual/model.onnx', target_platfo
72
modified = modify_onnx_change_op_type(model_path, modified_onnx_path, "CumSum", "CPUCumSum")
73
onnx_to_load = modified_onnx_path if modified else model_path
74
if modified:
75
- ret = rknn.reg_custom_op(RKCumSum())
+ ret = rknn.reg_custom_op(CPUCumSum())
76
77
if ret != 0:
78
raise RuntimeError("Register Custom OP failed!")
0 commit comments