Skip to content

Commit

Permalink
one one comment
Browse files Browse the repository at this point in the history
  • Loading branch information
xadupre committed Nov 14, 2023
1 parent e373a25 commit bd689c2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions onnxruntime/core/session/custom_ops.cc
Original file line number Diff line number Diff line change
Expand Up @@ -840,6 +840,7 @@ ONNX_NAMESPACE::OpSchema CreateSchema(const std::string& domain, const OrtCustom
const ONNX_NAMESPACE::TypeProto* type_proto = DataTypeImpl::TensorTypeFromONNXEnum(static_cast<int>(output_type))->GetTypeProto();
schema.TypeConstraint(output_name, {*ONNX_NAMESPACE::Utils::DataTypeUtils::ToType(*type_proto)}, "one type");
} else {
// Type inference fails if the schema does not return one unique option when IsHomogeneous is false.
schema.TypeConstraint(output_name, DataTypeImpl::ToString(SUPPORTED_TENSOR_TYPES), "all types");
}
}
Expand Down

0 comments on commit bd689c2

Please sign in to comment.