From 3f507fa9f812e92dfcd0bf0b050dd8311fe1ada8 Mon Sep 17 00:00:00 2001 From: Du Li Date: Tue, 12 Mar 2019 12:04:17 -0700 Subject: [PATCH] fix a shape inference bug. (#605) --- onnxruntime/core/graph/contrib_ops/contrib_defs.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)