vta.top.op

Namespace for supporting Relay operators on VTA.

Attributes

ENV

Functions

add_packed(cfg, lhs, rhs)
add_strategy_vta(attrs, inputs, out_type, target)
clip_strategy_vta(attrs, inputs, out_type, target)
compute_clip_vta(attrs, inputs, output_type)	Clip operator.
conv2d_strategy_vta(attrs, inputs, out_type, target)	conv2d vta strategy
conv2d_transpose_strategy_vta(attrs, inputs, out_type, ...)	conv2d_transpose vta strategy
dense_strategy_vta(attrs, inputs, out_type, target)	dense vta strategy
multiply_packed(cfg, lhs, rhs)
multiply_strategy_vta(attrs, inputs, out_type, target)
schedule_add_packed(cfg, outs)
schedule_alu_packed(cfg, outs)	alu packed schedule
schedule_multiply_packed(cfg, outs)

Module Contents

vta.top.op.add_packed(cfg, lhs, rhs)

vta.top.op.add_strategy_vta(attrs, inputs, out_type, target)

vta.top.op.clip_strategy_vta(attrs, inputs, out_type, target)

vta.top.op.compute_clip_vta(attrs, inputs, output_type)

Clip operator.

vta.top.op.conv2d_strategy_vta(attrs, inputs, out_type, target)

conv2d vta strategy

vta.top.op.conv2d_transpose_strategy_vta(attrs, inputs, out_type, target)

conv2d_transpose vta strategy

vta.top.op.dense_strategy_vta(attrs, inputs, out_type, target)

dense vta strategy

vta.top.op.multiply_packed(cfg, lhs, rhs)

vta.top.op.multiply_strategy_vta(attrs, inputs, out_type, target)

vta.top.op.schedule_add_packed(cfg, outs)

vta.top.op.schedule_alu_packed(cfg, outs)

alu packed schedule

vta.top.op.schedule_multiply_packed(cfg, outs)

vta.top.op.ENV