This is an automated email from the ASF dual-hosted git repository.
syfeng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 2afb854cf8 [Relay] fix the conflicted documentation description
(#15889)
2afb854cf8 is described below
commit 2afb854cf8bc31b0c5b4b100f97632546232773f
Author: Qingchao Shen <[email protected]>
AuthorDate: Thu Oct 12 18:04:02 2023 +0800
[Relay] fix the conflicted documentation description (#15889)
fix conflicted documentation description
---
python/tvm/relay/frontend/keras.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/python/tvm/relay/frontend/keras.py
b/python/tvm/relay/frontend/keras.py
index 6c82ebb427..e89c0a3c03 100644
--- a/python/tvm/relay/frontend/keras.py
+++ b/python/tvm/relay/frontend/keras.py
@@ -1426,9 +1426,9 @@ def from_keras(model, shape=None, layout="NCHW"):
Input shapes of the model, optional
layout: str
- One of 'NCHW' or 'NHWC', indicates how data should be arranged in
- the output model. Default layout is 'NCHW' as it in general
- performs better across TVM.
+ One of 'NWC', 'NCHW', 'NHWC', 'NDHWC' indicates how data should
+ be arranged in the output model. Default layout is 'NCHW' as it
+ in general performs better across TVM.
Returns
-------