add standard shape requirement

a884f396 · Shucai Xiao · 4bd1ee7a · a884f396
Commit a884f396 authored Jun 24, 2021 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/onnx/parse_generic_op.cpp src/onnx/parse_generic_op.cpp +1 -1

No files found.
--- a/src/onnx/parse_generic_op.cpp
+++ b/src/onnx/parse_generic_op.cpp
@@ -48,7 +48,7 @@ struct parse_generic_op : op_parser<parse_generic_op>

    bool needs_contiguous(const std::string& op_name) const
    {
-        return contains({"gather"}, op_name);
+        return contains({"gather", "scatter"}, op_name);
    }

    instruction_ref parse(const op_desc& opd,