saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix a shape inference bug. (#605)

Browse source
This commit is contained in:
Du Li 2019-03-12 12:04:17 -07:00 committed by Changming Sun
parent de9f1ff1ff
commit 3f507fa9f8
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/graph/contrib_ops/contrib_defs.cc
View file

@ -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)