vta.top.vta_group_conv2d

Group conv2D operator declaration and schedule registration for VTA.

Functions

group_conv2d_packed(cfg, data, kernel, strides, ...)	Packed group conv2d nchw function.
schedule_group_conv2d_packed(cfg, outs)	Schedule the packed conv2d.

Module Contents

vta.top.vta_group_conv2d.group_conv2d_packed(cfg, data, kernel, strides, padding, dilation, group, out_dtype)

Packed group conv2d nchw function.

vta.top.vta_group_conv2d.schedule_group_conv2d_packed(cfg, outs)

Schedule the packed conv2d.