File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -1951,11 +1951,13 @@ LogicalResult tosa::TableOp::inferReturnTypeComponents(
19511951}
19521952
19531953LogicalResult tosa::TableOp::verify () {
1954- TensorType inputType = getInput1 ().getType ();
1955- TensorType outputType = getOutput ().getType ();
1954+ const TensorType inputType = getInput1 ().getType ();
1955+ const TensorType outputType = getOutput ().getType ();
1956+
1957+ if (!inputType.hasRank () || !outputType.hasRank ())
1958+ return success ();
19561959
1957- if (inputType.hasRank () && outputType.hasRank () &&
1958- inputType.getRank () != outputType.getRank ())
1960+ if (inputType.getRank () != outputType.getRank ())
19591961 return emitOpError ()
19601962 << " expected input tensor rank to equal result tensor rank" ;
19611963
You can’t perform that action at this time.
0 commit comments