From c8f3742f0e56a9c13d1d86cac66c2e4ccb82cc35 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 8 Jul 2020 18:01:14 +0800 Subject: [PATCH] fix(mge/doc): fix wrong doc for megengine doc GitOrigin-RevId: a38dc8e10227c75c89e852611cc7c96f913bde2d --- python_module/megengine/jit/__init__.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python_module/megengine/jit/__init__.py b/python_module/megengine/jit/__init__.py index 5e2e495d3..6add694dc 100644 --- a/python_module/megengine/jit/__init__.py +++ b/python_module/megengine/jit/__init__.py @@ -455,17 +455,17 @@ class trace: :param enable_hwcd4: whether to use NHWCD4 data layout. This is faster on some OpenCL backend. - :param enable_nchw88: whether to use NCHW4 data layout. it currently + :param enable_nchw88: whether to use NCHW88 data layout. it currently used in X86 AVX backend. - :param enable_nchw44: whether to use NCHW4 data layout. it currently + :param enable_nchw44: whether to use NCHW44 data layout. it currently used in arm backend. - :param enable_nchw44_dot: whether to use NCHW4 data layout. it currently + :param enable_nchw44_dot: whether to use NCHW44_dot data layout. it currently used in armv8.2+dotprod backend. :param enable_nchw4: whether to use NCHW4 data layout. it currently used in nvidia backend(based on cudnn). - :param enable_nchw32 whether to use NCHW32 data layout. it currently + :param enable_nchw32: whether to use NCHW32 data layout. it currently used in nvidia backend with tensorcore(based on cudnn). - :param enable_chwn4 whether to use CHWN4 data layout. it currently + :param enable_chwn4: whether to use CHWN4 data layout. it currently used in nvidia backend with tensorcore. :param enable_fuse_conv_bias_nonlinearity: whether to fuse conv+bias+nonlinearty -- GitLab