vta.top.vta_conv2d_transpose

Conv2D_transpose operator declaration and schedule registration for VTA.

Module Contents

Functions

conv2d_transpose_packed(cfg, data, kernel, strides, ...)	Packed conv2d_transpose compute
schedule_conv2d_transpose_packed(cfg, outs)	Schedule packed conv2d_transpose

vta.top.vta_conv2d_transpose.conv2d_transpose_packed(cfg, data, kernel, strides, padding, out_dtype, output_padding=(0, 0))

Packed conv2d_transpose compute

vta.top.vta_conv2d_transpose.schedule_conv2d_transpose_packed(cfg, outs)

Schedule packed conv2d_transpose