提交 1529bce5 编写于 作者: M Megvii Engine Team

perf(opencl): add opencl weight transpose kernel

GitOrigin-RevId: 1102ecf4bb602f37d1335007bf40c4b145b0a4a4
上级 76cd4a52
......@@ -16,7 +16,7 @@ private:
public:
OprProxy() = default;
OprProxy(int k) : m_k{k} {}
void init(TopK*, const TensorLayoutArray&) {}
void init(TopK*, const TensorNDArray&) {}
void deduce_layout(TopK* opr, TensorLayoutArray& layouts) {
if (layouts.size() == 3) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册