diff --git a/onnxruntime/core/graph/contrib_ops/contrib_defs.cc b/onnxruntime/core/graph/contrib_ops/contrib_defs.cc index fb6eadcdae..287cea9a70 100644 --- a/onnxruntime/core/graph/contrib_ops/contrib_defs.cc +++ b/onnxruntime/core/graph/contrib_ops/contrib_defs.cc @@ -271,7 +271,7 @@ Sample echo operator.)DOC"); "T") .TypeConstraint("T", {"tensor(float)"}, "Constrain input0 and output types to float tensors") .TypeAndShapeInferenceFunction([](ONNX_NAMESPACE::InferenceContext& ctx) { - ONNX_NAMESPACE::convPoolTypeAndShapeInference(ctx, false, true); + ONNX_NAMESPACE::convPoolTypeAndShapeInference(ctx, true, false); }); ONNX_CONTRIB_OPERATOR_SCHEMA(FusedConv)