Skip to content

T
tic
Commit 8840b914 by Lianmin Zheng Committed by Tianqi Chen
Options

register conv2d_transpose as generic function (#1137)

parent 240be55b
Showing with 1 additions and 0 deletions
topi/python/topi/nn/conv2d_transpose.py
......@@ -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.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment