From 2afb854cf8bc31b0c5b4b100f97632546232773f Mon Sep 17 00:00:00 2001 From: Qingchao Shen Date: Thu, 12 Oct 2023 18:04:02 +0800 Subject: [PATCH] [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 6c82ebb427e7..e89c0a3c03a9 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 -------