Skip to content
Snippets Groups Projects
Commit 8840b914 authored by Lianmin Zheng's avatar Lianmin Zheng Committed by Tianqi Chen
Browse files

register conv2d_transpose as generic function (#1137)

parent 240be55b
No related branches found
No related tags found
No related merge requests found
......@@ -9,6 +9,7 @@ from .util import get_pad_tuple
from ..util import simplify
@tvm.target.generic_func
def conv2d_transpose_nchw(Input, Filter, strides, padding):
"""Transposed 2D convolution nchw forward operator.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment