vta.top.vta_dense

Dense operator declaration and schedule registration for VTA.

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.

Module Contents

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.