vta.top.vta_dense#

Dense operator declaration and schedule registration for VTA.

Module Contents#

Functions#

dense_packed(cfg, data, weight[, bias, out_dtype])

Dense function declaration.

is_packed_layout(layout)

Check if layout is packed layout

schedule_dense_packed(cfg, outs)

Packed dense schedule.

vta.top.vta_dense.dense_packed(cfg, data, weight, bias=None, out_dtype=None)[源代码]#

Dense function declaration.

vta.top.vta_dense.is_packed_layout(layout)[源代码]#

Check if layout is packed layout

vta.top.vta_dense.schedule_dense_packed(cfg, outs)[源代码]#

Packed dense schedule.