Add dense schedules to __init__ for cpu (apache#2855)

* Add dense schedules to __init__ for cpu * Add documentation for topi::shape * Add additional imports to topi CPU __init__.
MarisaKirisame · Apr 9, 2019 · 758a3b9 · 758a3b9
1 parent 2fc7662
commit 758a3b9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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