From 0b91fef712788d0c5a78c0c1a3dcb67bc163b09e Mon Sep 17 00:00:00 2001 From: Jhony Guan Date: Tue, 19 Mar 2019 11:53:36 -0700 Subject: [PATCH 1/3] Add dense schedules to __init__ for cpu --- topi/python/topi/x86/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/topi/python/topi/x86/__init__.py b/topi/python/topi/x86/__init__.py index 9e0e94e6cd2d..4d3fbab7dbdf 100644 --- a/topi/python/topi/x86/__init__.py +++ b/topi/python/topi/x86/__init__.py @@ -10,3 +10,4 @@ from .pooling import schedule_pool, schedule_global_pool from .bitserial_conv2d import schedule_bitserial_conv2d from .depthwise_conv2d import schedule_depthwise_conv2d_NCHWc +from .dense import _schedule_dense, _schedule_dense_pack, _schedule_dense_nopack From f8c58004c8db8ab2d1c497df9dea8be69f629049 Mon Sep 17 00:00:00 2001 From: Jhony Guan Date: Tue, 19 Mar 2019 12:02:51 -0700 Subject: [PATCH 2/3] Add documentation for topi::shape --- topi/include/topi/transform.h | 1 + 1 file changed, 1 insertion(+) diff --git a/topi/include/topi/transform.h b/topi/include/topi/transform.h index 57d442dc9206..464bd6facad5 100644 --- a/topi/include/topi/transform.h +++ b/topi/include/topi/transform.h @@ -1084,6 +1084,7 @@ inline Tensor layout_transform(const Tensor& src, /*! * \brief Get the shape of input tensor. * \param src the input tensor. + * \param dtype the type of the elements in the tensor. * \param name output tensor name. * \param tag output tensor tag. * \return Tensor of input shape. From 591f41c94f3654a0842af529c4ad1f4b9d5080de Mon Sep 17 00:00:00 2001 From: Jhony Guan Date: Mon, 25 Mar 2019 09:52:04 -0700 Subject: [PATCH 3/3] Add additional imports to topi CPU __init__. --- topi/python/topi/x86/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/topi/python/topi/x86/__init__.py b/topi/python/topi/x86/__init__.py index 4d3fbab7dbdf..638d428ec28a 100644 --- a/topi/python/topi/x86/__init__.py +++ b/topi/python/topi/x86/__init__.py @@ -11,3 +11,4 @@ from .bitserial_conv2d import schedule_bitserial_conv2d from .depthwise_conv2d import schedule_depthwise_conv2d_NCHWc from .dense import _schedule_dense, _schedule_dense_pack, _schedule_dense_nopack +from .batch_matmul import schedule_batch_matmul