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. diff --git a/topi/python/topi/x86/__init__.py b/topi/python/topi/x86/__init__.py index 9e0e94e6cd2d..638d428ec28a 100644 --- a/topi/python/topi/x86/__init__.py +++ b/topi/python/topi/x86/__init__.py @@ -10,3 +10,5 @@ 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 .batch_matmul import schedule_batch_matmul