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

Fix shape inference bug in GatherND contrib op (#1132)

Browse source
This commit is contained in:
Hariharan Seshadri 2019-05-30 17:05:44 -07:00 committed by GitHub
parent 4757933afe
commit facdf77f84
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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

@ -929,7 +929,7 @@ with the exception that numpy default keepdims to False instead of True.)DOC")
.TypeAndShapeInferenceFunction([](ONNX_NAMESPACE::InferenceContext& ctx) {
propagateElemTypeFromInputToOutput(ctx, 0, 0);
if (!hasNInputShapes(ctx, 2)) {
fail_shape_inference("GatherND requires two tensor inputs.");
return;
}
auto& data_shape = ctx.getInputType(0)->tensor_type().shape();
auto& indices_shape = ctx.getInputType(1)->tensor_type().shape();