Namespace for supporting Relay operators on VTA.
Module Contents
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)
|
|
Attributes
-
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[源代码]
