Skip if undefined data type

Signed-off-by: Aditya Goel <agoel4512@gmail.com>
onnx · Oct 16, 2023 · b64678c · b64678c
1 parent cd6531e
commit b64678c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnx/defs/traditionalml/defs.cc b/onnx/defs/traditionalml/defs.cc
@@ -392,7 +392,8 @@ ONNX_ML_OPERATOR_SET_SCHEMA(
           }
 
           auto default_attr = ctx.getAttribute("default_tensor");
-          if (nullptr != default_attr) {
+          if (nullptr != default_attr && default_attr->has_t() && default_attr->t().has_data_type() &&
+              default_attr->t().data_type() != TensorProto_DataType_UNDEFINED) {
             auto default_tensor = default_attr->t();
             if (default_tensor.data_type() != value_type) {
               fail_shape_inference(