Skip to content

Commit 2afb854

Browse files
authored
[Relay] fix the conflicted documentation description (apache#15889)
fix conflicted documentation description
1 parent ab1aef9 commit 2afb854

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/tvm/relay/frontend/keras.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1426,9 +1426,9 @@ def from_keras(model, shape=None, layout="NCHW"):
14261426
Input shapes of the model, optional
14271427
14281428
layout: str
1429-
One of 'NCHW' or 'NHWC', indicates how data should be arranged in
1430-
the output model. Default layout is 'NCHW' as it in general
1431-
performs better across TVM.
1429+
One of 'NWC', 'NCHW', 'NHWC', 'NDHWC' indicates how data should
1430+
be arranged in the output model. Default layout is 'NCHW' as it
1431+
in general performs better across TVM.
14321432
14331433
Returns
14341434
-------

0 commit comments

Comments
 (0)