vta.top.vta_group_conv2d#

Group conv2D operator declaration and schedule registration for VTA.

Module Contents#

Functions#

group_conv2d_packed(cfg, data, kernel, strides, ...)

Packed group conv2d nchw function.

schedule_group_conv2d_packed(cfg, outs)

Schedule the packed conv2d.

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.