File tree 2 files changed +6
-2
lines changed
2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -250,6 +250,8 @@ def _onnx_node_to_tensorflow_op(cls,
250
250
if handlers :
251
251
handler = handlers [node .domain ].get (node .op_type , None ) if node .domain in handlers else None
252
252
if handler :
253
+ if hasattr (handler , "WARNINGS" ):
254
+ common .logger .warning (handler .WARNINGS )
253
255
return handler .handle (node , tensor_dict = tensor_dict , strict = strict )
254
256
255
257
raise BackendIsNotSupposedToImplementIt ("{} is not implemented." .format (node .op_type ))
Original file line number Diff line number Diff line change @@ -29,11 +29,13 @@ def get_all_backend_handlers(opset_dict):
29
29
domain = handler .DOMAIN ,
30
30
max_inclusive_version = version ).since_version
31
31
except RuntimeError :
32
- common .logger .debug ("Fail to get since_version of {} in domain `{}` "
32
+ handler .WARNINGS = (((handler .WARNINGS + "\n " ) if hasattr (handler , "WARNINGS" ) else "" ) +
33
+ "Fail to get since_version of {} in domain `{}` "
33
34
"with max_inclusive_version={}. Set to 1." .format (
34
35
handler .ONNX_OP , handler .DOMAIN , version ))
35
36
else :
36
- common .logger .debug ("Unknown op {} in domain `{}`." .format (
37
+ handler .WARNINGS = (((handler .WARNINGS + "\n " ) if hasattr (handler , "WARNINGS" ) else "" ) +
38
+ "Unknown op {} in domain `{}`." .format (
37
39
handler .ONNX_OP , handler .DOMAIN or "ai.onnx" ))
38
40
handler .SINCE_VERSION = since_version
39
41
handlers .setdefault (domain , {})[handler .ONNX_OP ] = handler
You can’t perform that action at this time.
0 commit comments