vta.top.vta_conv2d

Conv2D operator declaration and schedule registration for VTA.

Functions

conv2d_packed(cfg, data, kernel, strides, padding, ...)	Packed conv2d function.
schedule_conv2d_packed(cfg, outs)	Schedule packed conv2d

Module Contents

vta.top.vta_conv2d.conv2d_packed(cfg, data, kernel, strides, padding, dilation, layout, out_dtype)

Packed conv2d function.

vta.top.vta_conv2d.schedule_conv2d_packed(cfg, outs)

Schedule packed conv2d