colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added api json for each op to support acuity (#596) 

Record constructor form of each operation as a json file to support acuity to call
timvx op

Type:  Documentation

Signed-off-by: Feiyue Chen <Feiyue.Chen@verisilicon.com>
This commit is contained in:
Chen Feiyue 2023-06-02 07:51:10 +08:00 committed by GitHub
parent ea8adc456a
commit aa7b3a6f8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
132 changed files with 2091 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
include/tim/vx/ops/BidirectionalSequenceLstm.json Executable file
View File

@ -0,0 +1,48 @@
{
"BidirectionalSequenceLstm":{
"parameters":
[
{"name": "cell_clip",
"dtype": "float"
},
{"name": "proj_clip",
"dtype": "float"
},
{"name": "act_type",
"dtype": "tim::vx::ops::BidirectionalSequenceLstm::ActivationType",
"range":["tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kNONE",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kRELU",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kRELU1",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kRELU6",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kTANH",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kSIGMOID",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kHARDSIGMOID"]
},
{"name": "forget_bias",
"dtype": "float"
},
{"name": "time_major",
"dtype": "bool",
"Optional": "true",
"default": "false"
},
{"name": "recurrent_act_type",
"dtype": "tim::vx::ops::BidirectionalSequenceLstm::ActivationType",
"Optional": "true",
"default": "tim::vx::ops::BidirectionalSequenceLstm::kSIGMOID",
"range":["tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kNONE",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kRELU",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kRELU1",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kRELU6",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kTANH",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kSIGMOID",
"tim::vx::ops::BidirectionalSequenceLstm::ActivationType::kHARDSIGMOID"]
},
{"name": "return_sequences",
"dtype": "bool",
"Optional": "true",
"default": "false"
}
]
}
}

27
include/tim/vx/ops/BidirectionalSequenceRNN.json Executable file
View File

@ -0,0 +1,27 @@
{
"BidirectionalSequenceRNN":{
"parameters":
[
{"name": "act_type",
"dtype": "tim::vx::ops::BidirectionalSequenceRnn::ActivationType",
"range":["tim::vx::ops::BidirectionalSequenceRnn::ActivationType::kNONE",
"tim::vx::ops::BidirectionalSequenceRnn::ActivationType::kRELU",
"tim::vx::ops::BidirectionalSequenceRnn::ActivationType::kRELU1",
"tim::vx::ops::BidirectionalSequenceRnn::ActivationType::kRELU6",
"tim::vx::ops::BidirectionalSequenceRnn::ActivationType::kTANH",
"tim::vx::ops::BidirectionalSequenceRnn::ActivationType::kSIGMOID",
"tim::vx::ops::BidirectionalSequenceRnn::ActivationType::kHARDSIGMOID"]
},
{"name": "time_major",
"dtype": "bool",
"Optional": "true",
"default": "false"
},
{"name": "merge_outputs",
"dtype": "bool",
"Optional": "true",
"default": "false"
}
]
}
}

6
include/tim/vx/ops/HashtableLookup.json Executable file
View File

@ -0,0 +1,6 @@
{
"HashtableLookup":{
"parameters":
[]
}
}

10
include/tim/vx/ops/L2Normalization.json Executable file
View File

@ -0,0 +1,10 @@
{
"L2Normalization":{
"parameters":
[
{"name": "axis",
"dtype": "int32_t"
}
]
}
}

17
include/tim/vx/ops/LayerNormalization.json Executable file
View File

@ -0,0 +1,17 @@
{
"LayerNormalization":{
"parameters":
[
{"name": "axis",
"dtype": "int32_t",
"Optional": "true",
"default": "0"
},
{"name": "eps",
"dtype": "float",
"Optional": "true",
"default": "1e-5f"
}
]
}
}

38
include/tim/vx/ops/MaxpoolWithArgmax.json Executable file
View File

@ -0,0 +1,38 @@
{
"MaxpoolWithArgmax":{
"parameters":
[
{"name": "padding",
"dtype": "tim::vx::PadType",
"range": ["tim::vx::PadType::NONE",
"tim::vx::PadType::AUTO",
"tim::vx::PadType::VALID",
"tim::vx::PadType::SAME"]
},
{"name": "ksize",
"dtype": "std::array<uint32_t,2>",
"Optional": "true",
"default": "{0,0}"
},
{"name": "stride",
"dtype": "std::array<uint32_t,2>"
},
{"name": "round_type",
"dtype": "tim::vx::RoundType",
"Optional": "true",
"default": "tim::vx::RoundType::FLOOR",
"range": ["tim::vx::RoundType::FLOOR",
"tim::vx::RoundType::CEILING"]
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

38
include/tim/vx/ops/MaxpoolWithArgmax2.json Executable file
View File

@ -0,0 +1,38 @@
{
"MaxpoolWithArgmax2":{
"parameters":
[
{"name": "padding",
"dtype": "tim::vx::PadType",
"range": ["tim::vx::PadType::NONE",
"tim::vx::PadType::AUTO",
"tim::vx::PadType::VALID",
"tim::vx::PadType::SAME"]
},
{"name": "ksize",
"dtype": "std::array<uint32_t,2>",
"Optional": "true",
"default": "{0,0}"
},
{"name": "stride",
"dtype": "std::array<uint32_t,2>"
},
{"name": "round_type",
"dtype": "tim::vx::RoundType",
"Optional": "true",
"default": "tim::vx::RoundType::FLOOR",
"range": ["tim::vx::RoundType::FLOOR",
"tim::vx::RoundType::CEILING"]
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

6
include/tim/vx/ops/abs.json Executable file
View File

@ -0,0 +1,6 @@
{
"Abs":{
"parameters":
[]
}
}

6
include/tim/vx/ops/add.json Executable file
View File

@ -0,0 +1,6 @@
{
"Add":{
"parameters":
[]
}
}

10
include/tim/vx/ops/addn.json Executable file
View File

@ -0,0 +1,10 @@
{
"AddN":{
"parameters":
[
{"name":"num_inputs",
"dtype": "uint32_t"
}
]
}
}

6
include/tim/vx/ops/and.json Executable file
View File

@ -0,0 +1,6 @@
{
"And":{
"parameters":
[]
}
}

10
include/tim/vx/ops/argmax.json Executable file
View File

@ -0,0 +1,10 @@
{
"ArgMax":{
"parameters":
[
{"name":"axis",
"dtype": "int32_t"
}
]
}
}

10
include/tim/vx/ops/argmin.json Executable file
View File

@ -0,0 +1,10 @@
{
"ArgMin":{
"parameters":
[
{"name":"axis",
"dtype": "int32_t"
}
]
}
}

21
include/tim/vx/ops/batch2space.json Executable file
View File

@ -0,0 +1,21 @@
{
"Batch2Space":{
"parameters":
[
{"name": "block_size",
"dtype": "std::vector<int>"
},
{"name": "crop",
"dtype": "std::vector<int>"
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

18
include/tim/vx/ops/batchnorm.json Executable file
View File

@ -0,0 +1,18 @@
{
"BatchNorm":{
"parameters":
[
{"name": "eps",
"dtype": "float"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

15
include/tim/vx/ops/broadcast.json Executable file
View File

@ -0,0 +1,15 @@
{
"Broadcast":{
"parameters":
[
{"name": "shape",
"dtype": "std::vector<int32_t>"
},
{"name": "dimensions",
"dtype": "std::vector<int32_t>",
"Optional": "true",
"default": "{}"
}
]
}
}

6
include/tim/vx/ops/cast.json Executable file
View File

@ -0,0 +1,6 @@
{
"Cast":{
"parameters":
[]
}
}

6
include/tim/vx/ops/ceil.json Executable file
View File

@ -0,0 +1,6 @@
{
"Ceil":{
"parameters":
[]
}
}

10
include/tim/vx/ops/celu.json Executable file
View File

@ -0,0 +1,10 @@
{
"Celu":{
"parameters":
[
{"name":"alpha",
"dtype": "float"
}
]
}
}

13
include/tim/vx/ops/clip.json Executable file
View File

@ -0,0 +1,13 @@
{
"Clip":{
"parameters":
[
{"name": "min",
"dtype": "float"
},
{"name": "max",
"dtype": "float"
}
]
}
}

13
include/tim/vx/ops/concat.json Executable file
View File

@ -0,0 +1,13 @@
{
"Concat":{
"parameters":
[
{"name": "axis",
"dtype": "uint32_t"
},
{"name": "input_cnt",
"dtype": "int"
}
]
}
}

32
include/tim/vx/ops/conv1d.json Executable file
View File

@ -0,0 +1,32 @@
{
"Conv1d":{
"parameters":
[
{"name": "pad",
"dtype": "std::array<uint32_t,2>"
},
{"name": "stride",
"dtype": "uint32_t"
},
{"name": "dilation",
"dtype": "uint32_t"
},
{"name": "multiplier",
"dtype": "int32_t",
"Optional":"true",
"default": "0"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WCN"
},
{"name": "kernel_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WIO"
}
]
}
}

58
include/tim/vx/ops/conv2d.json Executable file
View File

@ -0,0 +1,58 @@
{
"Conv2d":{
"parameters":
[
{"name": "weights",
"dtype": "int32_t",
"Optional":"true",
"default":"0"
},
{"name": "padding",
"dtype": "tim::vx::PadType",
"range": ["tim::vx::PadType::NONE",
"tim::vx::PadType::AUTO",
"tim::vx::PadType::VALID",
"tim::vx::PadType::SAME"]
},
{"name": "ksize",
"dtype": "std::array<uint32_t,2>",
"Optional": "true",
"default": "{0,0}"
},
{"name": "stride",
"dtype": "std::array<uint32_t,2>"
},
{"name": "dilation",
"dtype": "std::array<uint32_t,2>"
},
{"name": "pad",
"dtype": "std::array<uint32_t,4>",
"Optional":"true",
"default":"{0,0,0,0}"
},
{"name": "multiplier",
"dtype": "int32_t",
"Optional":"true",
"default": "0"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
},
{"name": "kernel_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHIO",
"range":["tim::vx::DataLayout::IWHO",
"tim::vx::DataLayout::OIWH",
"tim::vx::DataLayout::IOWH",
"tim::vx::DataLayout::WHIO"]
}
]
}
}

36
include/tim/vx/ops/conv3d.json Executable file
View File

@ -0,0 +1,36 @@
{
"Conv3d":{
"parameters":
[
{"name": "pad",
"dtype": "std::array<int32_t,6>"
},
{"name": "stride",
"dtype": "std::array<int32_t,3>"
},
{"name": "dilation",
"dtype": "std::array<int32_t,3>"
},
{"name": "multiplier",
"dtype": "int32_t",
"Optional":"true",
"default": "0"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHDCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHDCN",
"tim::vx::DataLayout::CWHDN"]
},
{"name": "kernel_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHDIO",
"range":["tim::vx::DataLayout::OIWHD",
"tim::vx::DataLayout::WHDIO"]
}
]
}
}

22
include/tim/vx/ops/cumsum.json Executable file
View File

@ -0,0 +1,22 @@
{
"CumSum":{
"parameters":
[
{"name": "axis",
"dtype": "int32_t",
"Optional":"true",
"default":"0"
},
{"name": "exclusive",
"dtype": "int32_t",
"Optional":"true",
"default":"0"
},
{"name": "reverse",
"dtype": "int32_t",
"Optional":"true",
"default":"0"
}
]
}
}

6
include/tim/vx/ops/dataconvert.json Executable file
View File

@ -0,0 +1,6 @@
{
"DataConvert":{
"parameters":
[]
}
}

38
include/tim/vx/ops/deconv1d.json Executable file
View File

@ -0,0 +1,38 @@
{
"DeConv1d":{
"parameters":
[
{"name": "pad",
"dtype": "std::array<uint32_t,2>"
},
{"name": "stride",
"dtype": "uint32_t"
},
{"name": "output_padding",
"dtype": "uint32_t"
},
{"name": "group",
"dtype": "uint32_t",
"Optional":"true",
"default":"1"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
},
{"name": "kernel_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHIO",
"range":["tim::vx::DataLayout::IWHO",
"tim::vx::DataLayout::OIWH",
"tim::vx::DataLayout::IOWH",
"tim::vx::DataLayout::WHIO"]
}
]
}
}

51
include/tim/vx/ops/deconv2d.json Executable file
View File

@ -0,0 +1,51 @@
{
"DeConv2d":{
"parameters":
[
{"name": "oc_count",
"dtype": "int32_t"
},
{"name": "pad_type",
"dtype": "tim::vx::PadType",
"range": ["tim::vx::PadType::NONE",
"tim::vx::PadType::AUTO",
"tim::vx::PadType::VALID",
"tim::vx::PadType::SAME"]
},
{"name": "ksize",
"dtype": "std::array<uint32_t,2>"
},
{"name": "stride",
"dtype": "std::array<uint32_t,2>"
},
{"name": "output_padding",
"dtype": "std::array<uint32_t,2>"
},
{"name": "pad",
"dtype": "std::array<uint32_t,4>"
},
{"name": "group",
"dtype": "uint32_t",
"Optional":"true",
"default":"1"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
},
{"name": "kernel_layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHIO",
"range":["tim::vx::DataLayout::IWHO",
"tim::vx::DataLayout::OIWH",
"tim::vx::DataLayout::IOWH",
"tim::vx::DataLayout::WHIO"]
}
]
}
}

25
include/tim/vx/ops/depth2space.json Executable file
View File

@ -0,0 +1,25 @@
{
"DepthToSpace":{
"parameters":
[
{"name": "block_size",
"dtype": "int"
},
{"name": "mode",
"Optional": "true",
"dtype": "tim::vx::ops::DepthToSpace::depth2space_mode",
"default": "tim::vx::ops::DepthToSpace::depth2space_mode::DCR_mode",
"range":["tim::vx::ops::DepthToSpace::depth2space_mode::DCR_mode",
"tim::vx::ops::DepthToSpace::depth2space_mode::CRD_mode"]
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

12
include/tim/vx/ops/div.json Executable file
View File

@ -0,0 +1,12 @@
{
"Div":{
"parameters":
[
{"name": "scale",
"dtype": "float",
"Optional": "true",
"default": "1.0"
}
]
}
}

10
include/tim/vx/ops/dropout.json Executable file
View File

@ -0,0 +1,10 @@
{
"Dropout":{
"parameters":
[
{"name": "ratio",
"dtype": "float"
}
]
}
}

12
include/tim/vx/ops/elu.json Executable file
View File

@ -0,0 +1,12 @@
{
"Elu":{
"parameters":
[
{"name":"alpha",
"dtype": "float",
"Optional":"true",
"default":"1"
}
]
}
}

6
include/tim/vx/ops/embeddinglookup.json Executable file
View File

@ -0,0 +1,6 @@
{
"EmbeddingLookup":{
"parameters":
[]
}
}

6
include/tim/vx/ops/equal.json Executable file
View File

@ -0,0 +1,6 @@
{
"Equal":{
"parameters":
[]
}
}

6
include/tim/vx/ops/erf.json Executable file
View File

@ -0,0 +1,6 @@
{
"Erf":{
"parameters":
[]
}
}

6
include/tim/vx/ops/exp.json Executable file
View File

@ -0,0 +1,6 @@
{
"Exp":{
"parameters":
[]
}
}

6
include/tim/vx/ops/floor.json Executable file
View File

@ -0,0 +1,6 @@
{
"Floor":{
"parameters":
[]
}
}

6
include/tim/vx/ops/floordiv.json Executable file
View File

@ -0,0 +1,6 @@
{
"FloorDiv":{
"parameters":
[]
}
}

15
include/tim/vx/ops/fullyconnected.json Executable file
View File

@ -0,0 +1,15 @@
{
"FullyConnected":{
"parameters":
[
{"name": "axis",
"dtype": "uint32_t"
},
{"name": "weights",
"dtype": "uint32_t",
"Optional": "true",
"default": "0"
}
]
}
}

15
include/tim/vx/ops/gather.json Executable file
View File

@ -0,0 +1,15 @@
{
"Gather":{
"parameters":
[
{"name": "axis",
"dtype": "int"
},
{"name": "batch_dims",
"dtype": "int",
"Optional": "true",
"default": "0"
}
]
}
}

10
include/tim/vx/ops/gatherelements.json Executable file
View File

@ -0,0 +1,10 @@
{
"GatherElements":{
"parameters":
[
{"name": "axis",
"dtype": "int32_t"
}
]
}
}

6
include/tim/vx/ops/gathernd.json Executable file
View File

@ -0,0 +1,6 @@
{
"GatherNd":{
"parameters":
[]
}
}

12
include/tim/vx/ops/gelu.json Executable file
View File

@ -0,0 +1,12 @@
{
"Gelu":{
"parameters":
[
{"name":"approximate",
"dtype": "bool",
"Optional":"true",
"default":"true"
}
]
}
}

6
include/tim/vx/ops/greater.json Executable file
View File

@ -0,0 +1,6 @@
{
"Greater":{
"parameters":
[]
}
}

6
include/tim/vx/ops/greaterOrEqual.json Executable file
View File

@ -0,0 +1,6 @@
{
"GreaterOrEqual":{
"parameters":
[]
}
}

39
include/tim/vx/ops/groupedConv1d.json Executable file
View File

@ -0,0 +1,39 @@
{
"GroupedConv1d":{
"parameters":
[
{"name": "padding",
"dtype": "tim::vx::PadType",
"range": ["tim::vx::PadType::NONE",
"tim::vx::PadType::AUTO",
"tim::vx::PadType::VALID",
"tim::vx::PadType::SAME"]
},
{"name": "pad",
"dtype": "std::array<uint32_t,2>",
"Optional":"true",
"default":"{0,0}"
},
{"name": "stride",
"dtype": "uint32_t"
},
{"name": "dilation",
"dtype": "uint32_t"
},
{"name": "group",
"dtype": "uint32_t"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WCN"
},
{"name": "kernel_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WIO"
}
]
}
}

37
include/tim/vx/ops/groupedConv2d.json Executable file
View File

@ -0,0 +1,37 @@
{
"GroupedConv2d":{
"parameters":
[
{"name": "pad",
"dtype": "std::array<uint32_t,4>"
},
{"name": "strides",
"dtype": "std::array<uint32_t,2>"
},
{"name": "dilation",
"dtype": "std::array<uint32_t,2>"
},
{"name": "group_number",
"dtype": "int32_t"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
},
{"name": "kernel_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHIO",
"range":["tim::vx::DataLayout::IWHO",
"tim::vx::DataLayout::OIWH",
"tim::vx::DataLayout::IOWH",
"tim::vx::DataLayout::WHIO"]
}
]
}
}

39
include/tim/vx/ops/grucell.json Executable file
View File

@ -0,0 +1,39 @@
{
"GRUCell":{
"parameters":
[
{"name": "num_units",
"dtype": "uint32_t"
},
{"name": "activation",
"dtype": "tim::vx::ops::GRUCell::ActivationType",
"Optional": "true",
"default": "tim::vx::ops::GRUCell::ActivationType::kTANH",
"range":["tim::vx::ops::GRUCell::ActivationType::kNONE",
"tim::vx::ops::GRUCell::ActivationType::kRELU",
"tim::vx::ops::GRUCell::ActivationType::kRELU1",
"tim::vx::ops::GRUCell::ActivationType::kRELU6",
"tim::vx::ops::GRUCell::ActivationType::kTANH",
"tim::vx::ops::GRUCell::ActivationType::kSIGMOID",
"tim::vx::ops::GRUCell::ActivationType::kHARDSIGMOID"]
},
{"name": "recurrent_act_type",
"dtype": "tim::vx::ops::GRUCell::ActivationType",
"Optional": "true",
"default": "tim::vx::ops::GRUCell::ActivationType::kSIGMOID",
"range":["tim::vx::ops::GRUCell::ActivationType::kNONE",
"tim::vx::ops::GRUCell::ActivationType::kRELU",
"tim::vx::ops::GRUCell::ActivationType::kRELU1",
"tim::vx::ops::GRUCell::ActivationType::kRELU6",
"tim::vx::ops::GRUCell::ActivationType::kTANH",
"tim::vx::ops::GRUCell::ActivationType::kSIGMOID",
"tim::vx::ops::GRUCell::ActivationType::kHARDSIGMOID"]
},
{"name": "reset_after",
"dtype": "vsi_bool",
"Optional": "true",
"default": "TRUE"
}
]
}
}

13
include/tim/vx/ops/hardsigmoid.json Executable file
View File

@ -0,0 +1,13 @@
{
"HardSigmoid":{
"parameters":
[
{"name":"alpha",
"dtype": "float"
},
{"name":"beta",
"dtype": "float"
}
]
}
}

6
include/tim/vx/ops/hardswish.json Executable file
View File

@ -0,0 +1,6 @@
{
"HardSwish":{
"parameters":
[]
}
}

12
include/tim/vx/ops/instanceNormalization.json Executable file
View File

@ -0,0 +1,12 @@
{
"InstanceNormalization":{
"parameters":
[
{"name": "eps",
"dtype": "float",
"Optional": "true",
"default": "1e-5f"
}
]
}
}

10
include/tim/vx/ops/leakyrelu.json Executable file
View File

@ -0,0 +1,10 @@
{
"LeakyRelu":{
"parameters":
[
{"name":"alpha",
"dtype": "float"
}
]
}
}

6
include/tim/vx/ops/less.json Executable file
View File

@ -0,0 +1,6 @@
{
"Less":{
"parameters":
[]
}
}

6
include/tim/vx/ops/lessOrEqual.json Executable file
View File

@ -0,0 +1,6 @@
{
"LessOrEqual":{
"parameters":
[]
}
}

15
include/tim/vx/ops/linear.json Executable file
View File

@ -0,0 +1,15 @@
{
"Linear":{
"parameters":
[
{"name":"a",
"dtype": "float"
},
{"name":"b",
"dtype": "float",
"Optional":"true",
"default":"0.0"
}
]
}
}

22
include/tim/vx/ops/localresponseNormalization.json Executable file
View File

@ -0,0 +1,22 @@
{
"LocalResponseNormalization":{
"parameters":
[
{"name": "size",
"dtype": "uint32_t"
},
{"name": "alpha",
"dtype": "float"
},
{"name": "beta",
"dtype": "float"
},
{"name": "bias",
"dtype": "float"
},
{"name": "axis",
"dtype": "int32_t"
}
]
}
}

6
include/tim/vx/ops/log.json Executable file
View File

@ -0,0 +1,6 @@
{
"Log":{
"parameters":
[]
}
}

6
include/tim/vx/ops/logicalnot.json Executable file
View File

@ -0,0 +1,6 @@
{
"LogicalNot":{
"parameters":
[]
}
}

6
include/tim/vx/ops/logicalor.json Executable file
View File

@ -0,0 +1,6 @@
{
"LogicalOr":{
"parameters":
[]
}
}

15
include/tim/vx/ops/logsoftmax.json Executable file
View File

@ -0,0 +1,15 @@
{
"LogSoftmax":{
"parameters":
[
{"name": "axis",
"dtype": "int32_t"
},
{"name": "beta",
"dtype": "float",
"Optional": "true",
"default": "1.0"
}
]
}
}

27
include/tim/vx/ops/matmul.json Executable file
View File

@ -0,0 +1,27 @@
{
"Matmul":{
"parameters":
[
{"name": "transpose_a",
"dtype": "bool",
"Optional": "true",
"default": "false"
},
{"name": "transpose_b",
"dtype": "bool",
"Optional": "true",
"default": "false"
},
{"name": "adjoint_a",
"dtype": "bool",
"Optional": "true",
"default": "false"
},
{"name": "adjoint_b",
"dtype": "bool",
"Optional": "true",
"default": "false"
}
]
}
}

6
include/tim/vx/ops/maximum.json Executable file
View File

@ -0,0 +1,6 @@
{
"Maximum":{
"parameters":
[]
}
}

36
include/tim/vx/ops/maxpool3d.json Executable file
View File

@ -0,0 +1,36 @@
{
"MaxPool3d":{
"parameters":
[
{"name": "round_type",
"dtype": "tim::vx::RoundType",
"range": ["tim::vx::RoundType::CEILING",
"tim::vx::RoundType::FLOOR"]
},
{"name": "ksize",
"dtype": "std::array<uint32_t,3>"
},
{"name": "stride",
"dtype": "std::array<uint32_t,3>"
},
{"name": "pad",
"dtype": "std::array<uint32_t,6>"
},
{"name": "pad_type",
"dtype": "tim::vx::PadType",
"range": ["tim::vx::PadType::NONE",
"tim::vx::PadType::AUTO",
"tim::vx::PadType::VALID",
"tim::vx::PadType::SAME"]
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHDCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHDCN",
"tim::vx::DataLayout::CWHDN"]
}
]
}
}

21
include/tim/vx/ops/maxunpool2d.json Executable file
View File

@ -0,0 +1,21 @@
{
"MaxUnpool2d":{
"parameters":
[
{"name": "ksize",
"dtype": "std::array<uint32_t,2>"
},
{"name": "stride",
"dtype": "std::array<uint32_t,2>"
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional":"true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

6
include/tim/vx/ops/minimum.json Executable file
View File

@ -0,0 +1,6 @@
{
"Minimum":{
"parameters":
[]
}
}

6
include/tim/vx/ops/mish.json Executable file
View File

@ -0,0 +1,6 @@
{
"Mish":{
"parameters":
[]
}
}

12
include/tim/vx/ops/mod.json Executable file
View File

@ -0,0 +1,12 @@
{
"Mod":{
"parameters":
[
{"name": "fmod",
"dtype": "int32_t",
"Optional": "true",
"default": "0"
}
]
}
}

15
include/tim/vx/ops/moments.json Executable file
View File

@ -0,0 +1,15 @@
{
"Moments":{
"parameters":
[
{"name": "axis",
"dtype": "std::vector<int32_t>"
},
{"name": "keep_dims",
"dtype": "bool",
"Optional": "true",
"default": "false"
}
]
}
}

12
include/tim/vx/ops/mul.json Executable file
View File

@ -0,0 +1,12 @@
{
"Multiply":{
"parameters":
[
{"name": "scale",
"dtype": "float",
"Optional":"true",
"default":"1.0"
}
]
}
}

16
include/tim/vx/ops/nbg.json Executable file
View File

@ -0,0 +1,16 @@
{
"NBG":{
"parameters":
[
{"name":"binary",
"dtype": "char*"
},
{"name": "input_count",
"dtype": "size_t"
},
{"name": "output_count",
"dtype": "size_t"
}
]
}
}

6
include/tim/vx/ops/neg.json Executable file
View File

@ -0,0 +1,6 @@
{
"Neg":{
"parameters":
[]
}
}

6
include/tim/vx/ops/notEqual.json Executable file
View File

@ -0,0 +1,6 @@
{
"NotEqual":{
"parameters":
[]
}
}

25
include/tim/vx/ops/onehot.json Executable file
View File

@ -0,0 +1,25 @@
{
"OneHot":{
"parameters":
[
{"name":"depth",
"dtype": "int32_t"
},
{"name": "on_value",
"dtype": "float",
"Optional": "true",
"default": "1"
},
{"name": "off_value",
"dtype": "float",
"Optional": "true",
"default": "0"
},
{"name": "axis",
"dtype": "int32_t",
"Optional": "true",
"default": "0"
}
]
}
}

25
include/tim/vx/ops/pad.json Executable file
View File

@ -0,0 +1,25 @@
{
"Pad":{
"parameters":
[
{"name":"front_size",
"dtype": "std::vector<uint32_t>"
},
{"name": "back_size",
"dtype": "std::vector<uint32_t>"
},
{"name": "const_val",
"dtype": "int32_t"
},
{"name": "pad_mode",
"dtype": "tim::vx::ops::Pad::pad_mode_type",
"Optional": "true",
"default": "tim::vx::ops::Pad::pad_mode_type::PAD_MODE_CONSTANT",
"range": ["tim::vx::ops::Pad::pad_mode_type::PAD_MODE_CONSTANT",
"tim::vx::ops::Pad::pad_mode_type::PAD_MODE_EDGE",
"tim::vx::ops::Pad::pad_mode_type::PAD_MODE_SYMMETRIC",
"tim::vx::ops::Pad::pad_mode_type::PAD_MODE_REFLECT"]
}
]
}
}

25
include/tim/vx/ops/padv2.json Executable file
View File

@ -0,0 +1,25 @@
{
"PadV2":{
"parameters":
[
{"name":"front_size",
"dtype": "std::vector<uint32_t>"
},
{"name": "back_size",
"dtype": "std::vector<uint32_t>"
},
{"name": "const_val",
"dtype": "float"
},
{"name": "pad_mode",
"dtype": "tim::vx::ops::PadV2::pad_mode_type",
"Optional": "true",
"default": "tim::vx::ops::PadV2::pad_mode_type::PAD_MODE_CONSTANT",
"range": ["tim::vx::ops::PadV2::pad_mode_type::PAD_MODE_CONSTANT",
"tim::vx::ops::PadV2::pad_mode_type::PAD_MODE_EDGE",
"tim::vx::ops::PadV2::pad_mode_type::PAD_MODE_SYMMETRIC",
"tim::vx::ops::PadV2::pad_mode_type::PAD_MODE_REFLECT"]
}
]
}
}

38
include/tim/vx/ops/pool1d.json Executable file
View File

@ -0,0 +1,38 @@
{
"Pool1d":{
"parameters":
[
{"name":"type",
"dtype": "tim::vx::PoolType",
"range": ["tim::vx::PoolType::MAX",
"tim::vx::PoolType::AVG",
"tim::vx::PoolType::L2",
"tim::vx::POOLType::AVG_ANDROID"]
},
{"name":"pad",
"dtype":"std::array<uint32_t, 2>"
},
{"name":"ksize",
"dtype":"uint32_t"
},
{"name":"stride",
"dtype":"uint32_t"
},
{"name":"round_type",
"dtype":"tim::vx::RoundType",
"Optional":"true",
"default":"tim::vx::RoundType::FLOOR",
"range": ["tim::vx::RoundType::FLOOR",
"tim::vx::RoundType::CEILING"]
},
{"name":"layout",
"dtype":"tim::vx::DataLayout",
"Optional":"true",
"default":"tim::vx::DataLayout::WCN",
"range": ["tim::vx::DataLayout::WCN",
"tim::vx::DataLayout::WIO"]
}
]
}
}

39
include/tim/vx/ops/pool2d.json Executable file
View File

@ -0,0 +1,39 @@
{
"Pool2d":{
"parameters":
[
{"name":"type",
"dtype": "tim::vx::PoolType",
"range": ["tim::vx::PoolType::MAX",
"tim::vx::PoolType::AVG",
"tim::vx::PoolType::L2",
"tim::vx::PoolType::AVG_ANDROID"]
},
{"name":"pad",
"dtype":"std::array<uint32_t, 4>"
},
{"name":"ksize",
"dtype":"std::array<uint32_t, 2>"
},
{"name":"stride",
"dtype":"std::array<uint32_t, 2>"
},
{"name":"round_type",
"dtype":"tim::vx::RoundType",
"Optional":"true",
"default":"tim::vx::RoundType::FLOOR",
"range": ["tim::vx::RoundType::FLOOR",
"tim::vx::RoundType::CEILING"]
},
{"name":"layout",
"dtype":"tim::vx::DataLayout",
"Optional":"true",
"default":"tim::vx::DataLayout::WHCN",
"range": ["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

6
include/tim/vx/ops/pow.json Executable file
View File

@ -0,0 +1,6 @@
{
"Pow":{
"parameters":
[]
}
}

10
include/tim/vx/ops/prelu.json Executable file
View File

@ -0,0 +1,10 @@
{
"Prelu":{
"parameters":
[
{"name":"axis",
"dtype": "int"
}
]
}
}

6
include/tim/vx/ops/rcp.json Executable file
View File

@ -0,0 +1,6 @@
{
"Rcp":{
"parameters":
[]
}
}

13
include/tim/vx/ops/reduceall.json Executable file
View File

@ -0,0 +1,13 @@
{
"ReduceAll":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
},
{"name":"keep_dims",
"dtype": "bool"
}
]
}
}

13
include/tim/vx/ops/reduceany.json Executable file
View File

@ -0,0 +1,13 @@
{
"ReduceAny":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
},
{"name":"keep_dims",
"dtype": "bool"
}
]
}
}

13
include/tim/vx/ops/reducemax.json Executable file
View File

@ -0,0 +1,13 @@
{
"ReduceMax":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
},
{"name":"keep_dims",
"dtype": "bool"
}
]
}
}

13
include/tim/vx/ops/reducemean.json Executable file
View File

@ -0,0 +1,13 @@
{
"ReduceMean":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
},
{"name":"keep_dims",
"dtype": "bool"
}
]
}
}

13
include/tim/vx/ops/reducemin.json Executable file
View File

@ -0,0 +1,13 @@
{
"ReduceMin":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
},
{"name":"keep_dims",
"dtype": "bool"
}
]
}
}

13
include/tim/vx/ops/reduceprod.json Executable file
View File

@ -0,0 +1,13 @@
{
"ReduceProd":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
},
{"name":"keep_dims",
"dtype": "bool"
}
]
}
}

13
include/tim/vx/ops/reducesum.json Executable file
View File

@ -0,0 +1,13 @@
{
"ReduceSum":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
},
{"name":"keep_dims",
"dtype": "bool"
}
]
}
}

6
include/tim/vx/ops/relu.json Executable file
View File

@ -0,0 +1,6 @@
{
"Relu":{
"parameters":
[]
}
}

6
include/tim/vx/ops/relu1.json Executable file
View File

@ -0,0 +1,6 @@
{
"Relu1":{
"parameters":
[]
}
}

6
include/tim/vx/ops/relu6.json Executable file
View File

@ -0,0 +1,6 @@
{
"Relu6":{
"parameters":
[]
}
}

10
include/tim/vx/ops/reorg.json Executable file
View File

@ -0,0 +1,10 @@
{
"Reorg":{
"parameters":
[
{"name":"stride",
"dtype": "uint32_t"
}
]
}
}

10
include/tim/vx/ops/reshape.json Executable file
View File

@ -0,0 +1,10 @@
{
"Reshape":{
"parameters":
[
{"name": "size",
"dtype": "std::vector<uint32_t>"
}
]
}
}

36
include/tim/vx/ops/resize.json Executable file
View File

@ -0,0 +1,36 @@
{
"Resize":{
"parameters":
[
{"name":"type",
"dtype": "tim::vx::ResizeType",
"range": ["tim::vx::ResizeType::NEAREST_NEIGHBOR",
"tim::vx::ResizeType::BILINEAR",
"tim::vx::ResizeType::AREA"]
},
{"name": "factor",
"dtype": "float"
},
{"name": "align_corners",
"dtype": "bool"
},
{"name": "half_pixel_centers",
"dtype": "bool"
},
{"name": "target_height",
"dtype": "int"
},
{"name": "target_width",
"dtype": "int"
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

33
include/tim/vx/ops/resize1d.json Executable file
View File

@ -0,0 +1,33 @@
{
"Resize1d":{
"parameters":
[
{"name":"type",
"dtype": "tim::vx::ResizeType",
"range": ["tim::vx::ResizeType::NEAREST_NEIGHBOR",
"tim::vx::ResizeType::BILINEAR",
"tim::vx::ResizeType::AREA"]
},
{"name": "factor",
"dtype": "float"
},
{"name": "align_corners",
"dtype": "bool"
},
{"name": "half_pixel_centers",
"dtype": "bool"
},
{"name": "target_size",
"dtype": "int"
},
{"name": "layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}

10
include/tim/vx/ops/reverse.json Executable file
View File

@ -0,0 +1,10 @@
{
"Reverse":{
"parameters":
[
{"name":"axis",
"dtype": "std::vector<int32_t>"
}
]
}
}

17
include/tim/vx/ops/rnncell.json Executable file
View File

@ -0,0 +1,17 @@
{
"RNNCell":{
"parameters":
[
{"name": "activation",
"dtype": "tim::vx::ops::RNNCell::ActivationType",
"range": ["tim::vx::ops::RNNCell::ActivationType::kNONE",
"tim::vx::ops::RNNCell::ActivationType::kRELU",
"tim::vx::ops::RNNCell::ActivationType::kRELU1",
"tim::vx::ops::RNNCell::ActivationType::kRELU6",
"tim::vx::ops::RNNCell::ActivationType::kTANH",
"tim::vx::ops::RNNCell::ActivationType::kSIGMOID",
"tim::vx::ops::RNNCell::ActivationType::kHARDSIGMOID"]
}
]
}
}

25
include/tim/vx/ops/roialign.json Executable file
View File

@ -0,0 +1,25 @@
{
"RoiAlign":{
"parameters":
[
{"name":"output_height",
"dtype": "int32_t"
},
{"name":"output_weight",
"dtype": "int32_t"
},
{"name":"height_ratio",
"dtype": "float"
},
{"name":"width_ratio",
"dtype": "float"
},
{"name":"height_sample_num",
"dtype": "int32_t"
},
{"name":"width_sample_num",
"dtype": "int32_t"
}
]
}
}

20
include/tim/vx/ops/roipool.json Executable file
View File

@ -0,0 +1,20 @@
{
"RoiPool":{
"parameters":
[
{"name":"type",
"dtype": "tim::vx::PoolType",
"range": ["tim::vx::PoolType::MAX",
"tim::vx::PoolType::AVG",
"tim::vx::PoolType::L2",
"tim::vx::POOLType::AVG_ANDROID"]
},
{"name":"scale",
"dtype": "float"
},
{"name":"size",
"dtype": "std::array<uint32_t, 2>"
}
]
}
}

6
include/tim/vx/ops/round.json Executable file
View File

@ -0,0 +1,6 @@
{
"Round":{
"parameters":
[]
}
}

6
include/tim/vx/ops/rsqrt.json Executable file
View File

@ -0,0 +1,6 @@
{
"Rsqrt":{
"parameters":
[]
}
}

Some files were not shown because too many files have changed in this diff Show More