提交 bfee46f8 编写于 作者: Daya Khudia's avatar Daya Khudia 提交者: Facebook Github Bot

Update argument list for non-fbgemm path for qconv_prepack (#23521)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23521

non-fbgemm path should have the same arguments as fbgemm path.

Reviewed By: jianyuh

Differential Revision: D16547637

fbshipit-source-id: bb00d725fb968cbee32defb8facd2799a7e79bb4
上级 65a89472
......@@ -85,6 +85,9 @@ class QConvPackWeightInt8 final : public c10::OperatorKernel {
#else // USE_FBGEMM
Tensor operator()(
Tensor, /* weight */
torch::List<int64_t>, /* stride */
torch::List<int64_t>, /* padding */
torch::List<int64_t>, /* dilation */
int64_t /* groups */
) {
TORCH_CHECK(
......
Markdown 格式
0% or
您添加了 0 到此讨论。请谨慎行事。
先完成此消息的编辑!
想要评论请 注册