diff --git a/include/tim/vx/ops/BidirectionalSequenceLstm.json b/include/tim/vx/ops/BidirectionalSequenceLstm.json new file mode 100755 index 0000000..2f8465d --- /dev/null +++ b/include/tim/vx/ops/BidirectionalSequenceLstm.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/BidirectionalSequenceRNN.json b/include/tim/vx/ops/BidirectionalSequenceRNN.json new file mode 100755 index 0000000..084d7b3 --- /dev/null +++ b/include/tim/vx/ops/BidirectionalSequenceRNN.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/HashtableLookup.json b/include/tim/vx/ops/HashtableLookup.json new file mode 100755 index 0000000..1acb720 --- /dev/null +++ b/include/tim/vx/ops/HashtableLookup.json @@ -0,0 +1,6 @@ +{ + "HashtableLookup":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/L2Normalization.json b/include/tim/vx/ops/L2Normalization.json new file mode 100755 index 0000000..bed979c --- /dev/null +++ b/include/tim/vx/ops/L2Normalization.json @@ -0,0 +1,10 @@ +{ + "L2Normalization":{ + "parameters": + [ + {"name": "axis", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/LayerNormalization.json b/include/tim/vx/ops/LayerNormalization.json new file mode 100755 index 0000000..53d685b --- /dev/null +++ b/include/tim/vx/ops/LayerNormalization.json @@ -0,0 +1,17 @@ +{ + "LayerNormalization":{ + "parameters": + [ + {"name": "axis", + "dtype": "int32_t", + "Optional": "true", + "default": "0" + }, + {"name": "eps", + "dtype": "float", + "Optional": "true", + "default": "1e-5f" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/MaxpoolWithArgmax.json b/include/tim/vx/ops/MaxpoolWithArgmax.json new file mode 100755 index 0000000..56699eb --- /dev/null +++ b/include/tim/vx/ops/MaxpoolWithArgmax.json @@ -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", + "Optional": "true", + "default": "{0,0}" + }, + {"name": "stride", + "dtype": "std::array" + }, + {"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"] + } + + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/MaxpoolWithArgmax2.json b/include/tim/vx/ops/MaxpoolWithArgmax2.json new file mode 100755 index 0000000..5554537 --- /dev/null +++ b/include/tim/vx/ops/MaxpoolWithArgmax2.json @@ -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", + "Optional": "true", + "default": "{0,0}" + }, + {"name": "stride", + "dtype": "std::array" + }, + {"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"] + } + + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/abs.json b/include/tim/vx/ops/abs.json new file mode 100755 index 0000000..28301d6 --- /dev/null +++ b/include/tim/vx/ops/abs.json @@ -0,0 +1,6 @@ +{ + "Abs":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/add.json b/include/tim/vx/ops/add.json new file mode 100755 index 0000000..f16f15a --- /dev/null +++ b/include/tim/vx/ops/add.json @@ -0,0 +1,6 @@ +{ + "Add":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/addn.json b/include/tim/vx/ops/addn.json new file mode 100755 index 0000000..2c684f5 --- /dev/null +++ b/include/tim/vx/ops/addn.json @@ -0,0 +1,10 @@ +{ + "AddN":{ + "parameters": + [ + {"name":"num_inputs", + "dtype": "uint32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/and.json b/include/tim/vx/ops/and.json new file mode 100755 index 0000000..c4f09f6 --- /dev/null +++ b/include/tim/vx/ops/and.json @@ -0,0 +1,6 @@ +{ + "And":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/argmax.json b/include/tim/vx/ops/argmax.json new file mode 100755 index 0000000..02b12b7 --- /dev/null +++ b/include/tim/vx/ops/argmax.json @@ -0,0 +1,10 @@ +{ + "ArgMax":{ + "parameters": + [ + {"name":"axis", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/argmin.json b/include/tim/vx/ops/argmin.json new file mode 100755 index 0000000..39505e8 --- /dev/null +++ b/include/tim/vx/ops/argmin.json @@ -0,0 +1,10 @@ +{ + "ArgMin":{ + "parameters": + [ + {"name":"axis", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/batch2space.json b/include/tim/vx/ops/batch2space.json new file mode 100755 index 0000000..75e8880 --- /dev/null +++ b/include/tim/vx/ops/batch2space.json @@ -0,0 +1,21 @@ +{ + "Batch2Space":{ + "parameters": + [ + {"name": "block_size", + "dtype": "std::vector" + }, + {"name": "crop", + "dtype": "std::vector" + }, + {"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"] + } + ] + } +} diff --git a/include/tim/vx/ops/batchnorm.json b/include/tim/vx/ops/batchnorm.json new file mode 100755 index 0000000..f608ac7 --- /dev/null +++ b/include/tim/vx/ops/batchnorm.json @@ -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"] + } + ] + } +} diff --git a/include/tim/vx/ops/broadcast.json b/include/tim/vx/ops/broadcast.json new file mode 100755 index 0000000..ef8ebaa --- /dev/null +++ b/include/tim/vx/ops/broadcast.json @@ -0,0 +1,15 @@ +{ + "Broadcast":{ + "parameters": + [ + {"name": "shape", + "dtype": "std::vector" + }, + {"name": "dimensions", + "dtype": "std::vector", + "Optional": "true", + "default": "{}" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/cast.json b/include/tim/vx/ops/cast.json new file mode 100755 index 0000000..c6ee83e --- /dev/null +++ b/include/tim/vx/ops/cast.json @@ -0,0 +1,6 @@ +{ + "Cast":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/ceil.json b/include/tim/vx/ops/ceil.json new file mode 100755 index 0000000..6cf1ec0 --- /dev/null +++ b/include/tim/vx/ops/ceil.json @@ -0,0 +1,6 @@ +{ + "Ceil":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/celu.json b/include/tim/vx/ops/celu.json new file mode 100755 index 0000000..a66e9b0 --- /dev/null +++ b/include/tim/vx/ops/celu.json @@ -0,0 +1,10 @@ +{ + "Celu":{ + "parameters": + [ + {"name":"alpha", + "dtype": "float" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/clip.json b/include/tim/vx/ops/clip.json new file mode 100755 index 0000000..dc61cfc --- /dev/null +++ b/include/tim/vx/ops/clip.json @@ -0,0 +1,13 @@ +{ + "Clip":{ + "parameters": + [ + {"name": "min", + "dtype": "float" + }, + {"name": "max", + "dtype": "float" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/concat.json b/include/tim/vx/ops/concat.json new file mode 100755 index 0000000..eb51331 --- /dev/null +++ b/include/tim/vx/ops/concat.json @@ -0,0 +1,13 @@ +{ + "Concat":{ + "parameters": + [ + {"name": "axis", + "dtype": "uint32_t" + }, + {"name": "input_cnt", + "dtype": "int" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/conv1d.json b/include/tim/vx/ops/conv1d.json new file mode 100755 index 0000000..cfa3498 --- /dev/null +++ b/include/tim/vx/ops/conv1d.json @@ -0,0 +1,32 @@ +{ + "Conv1d":{ + "parameters": + + [ + {"name": "pad", + "dtype": "std::array" + }, + {"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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/conv2d.json b/include/tim/vx/ops/conv2d.json new file mode 100755 index 0000000..988f01f --- /dev/null +++ b/include/tim/vx/ops/conv2d.json @@ -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", + "Optional": "true", + "default": "{0,0}" + }, + {"name": "stride", + "dtype": "std::array" + }, + {"name": "dilation", + "dtype": "std::array" + }, + {"name": "pad", + "dtype": "std::array", + "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"] + } + + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/conv3d.json b/include/tim/vx/ops/conv3d.json new file mode 100755 index 0000000..0634de8 --- /dev/null +++ b/include/tim/vx/ops/conv3d.json @@ -0,0 +1,36 @@ +{ + "Conv3d":{ + "parameters": + [ + {"name": "pad", + "dtype": "std::array" + }, + {"name": "stride", + "dtype": "std::array" + }, + {"name": "dilation", + "dtype": "std::array" + }, + {"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"] + } + ] + } +} diff --git a/include/tim/vx/ops/cumsum.json b/include/tim/vx/ops/cumsum.json new file mode 100755 index 0000000..397fe3e --- /dev/null +++ b/include/tim/vx/ops/cumsum.json @@ -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" + } + ] + } +} diff --git a/include/tim/vx/ops/dataconvert.json b/include/tim/vx/ops/dataconvert.json new file mode 100755 index 0000000..527eba6 --- /dev/null +++ b/include/tim/vx/ops/dataconvert.json @@ -0,0 +1,6 @@ +{ + "DataConvert":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/deconv1d.json b/include/tim/vx/ops/deconv1d.json new file mode 100755 index 0000000..2ec5397 --- /dev/null +++ b/include/tim/vx/ops/deconv1d.json @@ -0,0 +1,38 @@ +{ + "DeConv1d":{ + "parameters": + [ + {"name": "pad", + "dtype": "std::array" + }, + {"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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/deconv2d.json b/include/tim/vx/ops/deconv2d.json new file mode 100755 index 0000000..a3fb7b0 --- /dev/null +++ b/include/tim/vx/ops/deconv2d.json @@ -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" + }, + {"name": "stride", + "dtype": "std::array" + }, + {"name": "output_padding", + "dtype": "std::array" + }, + {"name": "pad", + "dtype": "std::array" + }, + {"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"] + } + ] + } +} diff --git a/include/tim/vx/ops/depth2space.json b/include/tim/vx/ops/depth2space.json new file mode 100755 index 0000000..366278e --- /dev/null +++ b/include/tim/vx/ops/depth2space.json @@ -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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/div.json b/include/tim/vx/ops/div.json new file mode 100755 index 0000000..d6781f3 --- /dev/null +++ b/include/tim/vx/ops/div.json @@ -0,0 +1,12 @@ +{ + "Div":{ + "parameters": + [ + {"name": "scale", + "dtype": "float", + "Optional": "true", + "default": "1.0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/dropout.json b/include/tim/vx/ops/dropout.json new file mode 100755 index 0000000..aec95a4 --- /dev/null +++ b/include/tim/vx/ops/dropout.json @@ -0,0 +1,10 @@ +{ + "Dropout":{ + "parameters": + [ + {"name": "ratio", + "dtype": "float" + } + ] + } +} diff --git a/include/tim/vx/ops/elu.json b/include/tim/vx/ops/elu.json new file mode 100755 index 0000000..47a078c --- /dev/null +++ b/include/tim/vx/ops/elu.json @@ -0,0 +1,12 @@ +{ + "Elu":{ + "parameters": + [ + {"name":"alpha", + "dtype": "float", + "Optional":"true", + "default":"1" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/embeddinglookup.json b/include/tim/vx/ops/embeddinglookup.json new file mode 100755 index 0000000..114cbe0 --- /dev/null +++ b/include/tim/vx/ops/embeddinglookup.json @@ -0,0 +1,6 @@ +{ + "EmbeddingLookup":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/equal.json b/include/tim/vx/ops/equal.json new file mode 100755 index 0000000..7d318dc --- /dev/null +++ b/include/tim/vx/ops/equal.json @@ -0,0 +1,6 @@ +{ + "Equal":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/erf.json b/include/tim/vx/ops/erf.json new file mode 100755 index 0000000..e33477a --- /dev/null +++ b/include/tim/vx/ops/erf.json @@ -0,0 +1,6 @@ +{ + "Erf":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/exp.json b/include/tim/vx/ops/exp.json new file mode 100755 index 0000000..b81919a --- /dev/null +++ b/include/tim/vx/ops/exp.json @@ -0,0 +1,6 @@ +{ + "Exp":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/floor.json b/include/tim/vx/ops/floor.json new file mode 100755 index 0000000..7ef7d6e --- /dev/null +++ b/include/tim/vx/ops/floor.json @@ -0,0 +1,6 @@ +{ + "Floor":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/floordiv.json b/include/tim/vx/ops/floordiv.json new file mode 100755 index 0000000..61332a6 --- /dev/null +++ b/include/tim/vx/ops/floordiv.json @@ -0,0 +1,6 @@ +{ + "FloorDiv":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/fullyconnected.json b/include/tim/vx/ops/fullyconnected.json new file mode 100755 index 0000000..3c9588e --- /dev/null +++ b/include/tim/vx/ops/fullyconnected.json @@ -0,0 +1,15 @@ +{ + "FullyConnected":{ + "parameters": + [ + {"name": "axis", + "dtype": "uint32_t" + }, + {"name": "weights", + "dtype": "uint32_t", + "Optional": "true", + "default": "0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/gather.json b/include/tim/vx/ops/gather.json new file mode 100755 index 0000000..20a1179 --- /dev/null +++ b/include/tim/vx/ops/gather.json @@ -0,0 +1,15 @@ +{ + "Gather":{ + "parameters": + [ + {"name": "axis", + "dtype": "int" + }, + {"name": "batch_dims", + "dtype": "int", + "Optional": "true", + "default": "0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/gatherelements.json b/include/tim/vx/ops/gatherelements.json new file mode 100755 index 0000000..46a4d3e --- /dev/null +++ b/include/tim/vx/ops/gatherelements.json @@ -0,0 +1,10 @@ +{ + "GatherElements":{ + "parameters": + [ + {"name": "axis", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/gathernd.json b/include/tim/vx/ops/gathernd.json new file mode 100755 index 0000000..c5d760b --- /dev/null +++ b/include/tim/vx/ops/gathernd.json @@ -0,0 +1,6 @@ +{ + "GatherNd":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/gelu.json b/include/tim/vx/ops/gelu.json new file mode 100755 index 0000000..fd0b23e --- /dev/null +++ b/include/tim/vx/ops/gelu.json @@ -0,0 +1,12 @@ +{ + "Gelu":{ + "parameters": + [ + {"name":"approximate", + "dtype": "bool", + "Optional":"true", + "default":"true" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/greater.json b/include/tim/vx/ops/greater.json new file mode 100755 index 0000000..ab7ff4d --- /dev/null +++ b/include/tim/vx/ops/greater.json @@ -0,0 +1,6 @@ +{ + "Greater":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/greaterOrEqual.json b/include/tim/vx/ops/greaterOrEqual.json new file mode 100755 index 0000000..122526b --- /dev/null +++ b/include/tim/vx/ops/greaterOrEqual.json @@ -0,0 +1,6 @@ +{ + "GreaterOrEqual":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/groupedConv1d.json b/include/tim/vx/ops/groupedConv1d.json new file mode 100755 index 0000000..a49813d --- /dev/null +++ b/include/tim/vx/ops/groupedConv1d.json @@ -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", + "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" + } + + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/groupedConv2d.json b/include/tim/vx/ops/groupedConv2d.json new file mode 100755 index 0000000..5a27c19 --- /dev/null +++ b/include/tim/vx/ops/groupedConv2d.json @@ -0,0 +1,37 @@ +{ + "GroupedConv2d":{ + "parameters": + [ + {"name": "pad", + "dtype": "std::array" + }, + {"name": "strides", + "dtype": "std::array" + }, + {"name": "dilation", + "dtype": "std::array" + }, + {"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"] + } + + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/grucell.json b/include/tim/vx/ops/grucell.json new file mode 100755 index 0000000..ba54f51 --- /dev/null +++ b/include/tim/vx/ops/grucell.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/hardsigmoid.json b/include/tim/vx/ops/hardsigmoid.json new file mode 100755 index 0000000..2582753 --- /dev/null +++ b/include/tim/vx/ops/hardsigmoid.json @@ -0,0 +1,13 @@ +{ + "HardSigmoid":{ + "parameters": + [ + {"name":"alpha", + "dtype": "float" + }, + {"name":"beta", + "dtype": "float" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/hardswish.json b/include/tim/vx/ops/hardswish.json new file mode 100755 index 0000000..d4f766c --- /dev/null +++ b/include/tim/vx/ops/hardswish.json @@ -0,0 +1,6 @@ +{ + "HardSwish":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/instanceNormalization.json b/include/tim/vx/ops/instanceNormalization.json new file mode 100755 index 0000000..0361382 --- /dev/null +++ b/include/tim/vx/ops/instanceNormalization.json @@ -0,0 +1,12 @@ +{ + "InstanceNormalization":{ + "parameters": + [ + {"name": "eps", + "dtype": "float", + "Optional": "true", + "default": "1e-5f" + } + ] + } +} diff --git a/include/tim/vx/ops/leakyrelu.json b/include/tim/vx/ops/leakyrelu.json new file mode 100755 index 0000000..5e78229 --- /dev/null +++ b/include/tim/vx/ops/leakyrelu.json @@ -0,0 +1,10 @@ +{ + "LeakyRelu":{ + "parameters": + [ + {"name":"alpha", + "dtype": "float" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/less.json b/include/tim/vx/ops/less.json new file mode 100755 index 0000000..fd9fb92 --- /dev/null +++ b/include/tim/vx/ops/less.json @@ -0,0 +1,6 @@ +{ + "Less":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/lessOrEqual.json b/include/tim/vx/ops/lessOrEqual.json new file mode 100755 index 0000000..e92533e --- /dev/null +++ b/include/tim/vx/ops/lessOrEqual.json @@ -0,0 +1,6 @@ +{ + "LessOrEqual":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/linear.json b/include/tim/vx/ops/linear.json new file mode 100755 index 0000000..fb56533 --- /dev/null +++ b/include/tim/vx/ops/linear.json @@ -0,0 +1,15 @@ +{ + "Linear":{ + "parameters": + [ + {"name":"a", + "dtype": "float" + }, + {"name":"b", + "dtype": "float", + "Optional":"true", + "default":"0.0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/localresponseNormalization.json b/include/tim/vx/ops/localresponseNormalization.json new file mode 100755 index 0000000..6f375dd --- /dev/null +++ b/include/tim/vx/ops/localresponseNormalization.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/log.json b/include/tim/vx/ops/log.json new file mode 100755 index 0000000..84beeab --- /dev/null +++ b/include/tim/vx/ops/log.json @@ -0,0 +1,6 @@ +{ + "Log":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/logicalnot.json b/include/tim/vx/ops/logicalnot.json new file mode 100755 index 0000000..20d70d4 --- /dev/null +++ b/include/tim/vx/ops/logicalnot.json @@ -0,0 +1,6 @@ +{ + "LogicalNot":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/logicalor.json b/include/tim/vx/ops/logicalor.json new file mode 100755 index 0000000..1bd7cdf --- /dev/null +++ b/include/tim/vx/ops/logicalor.json @@ -0,0 +1,6 @@ +{ + "LogicalOr":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/logsoftmax.json b/include/tim/vx/ops/logsoftmax.json new file mode 100755 index 0000000..0a266ef --- /dev/null +++ b/include/tim/vx/ops/logsoftmax.json @@ -0,0 +1,15 @@ +{ + "LogSoftmax":{ + "parameters": + [ + {"name": "axis", + "dtype": "int32_t" + }, + {"name": "beta", + "dtype": "float", + "Optional": "true", + "default": "1.0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/matmul.json b/include/tim/vx/ops/matmul.json new file mode 100755 index 0000000..fa89c4b --- /dev/null +++ b/include/tim/vx/ops/matmul.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/maximum.json b/include/tim/vx/ops/maximum.json new file mode 100755 index 0000000..08e33fe --- /dev/null +++ b/include/tim/vx/ops/maximum.json @@ -0,0 +1,6 @@ +{ + "Maximum":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/maxpool3d.json b/include/tim/vx/ops/maxpool3d.json new file mode 100755 index 0000000..9195899 --- /dev/null +++ b/include/tim/vx/ops/maxpool3d.json @@ -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" + }, + {"name": "stride", + "dtype": "std::array" + }, + {"name": "pad", + "dtype": "std::array" + }, + {"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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/maxunpool2d.json b/include/tim/vx/ops/maxunpool2d.json new file mode 100755 index 0000000..4ef22b8 --- /dev/null +++ b/include/tim/vx/ops/maxunpool2d.json @@ -0,0 +1,21 @@ +{ + "MaxUnpool2d":{ + "parameters": + [ + {"name": "ksize", + "dtype": "std::array" + }, + {"name": "stride", + "dtype": "std::array" + }, + {"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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/minimum.json b/include/tim/vx/ops/minimum.json new file mode 100755 index 0000000..206b1c3 --- /dev/null +++ b/include/tim/vx/ops/minimum.json @@ -0,0 +1,6 @@ +{ + "Minimum":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/mish.json b/include/tim/vx/ops/mish.json new file mode 100755 index 0000000..b9ea0ea --- /dev/null +++ b/include/tim/vx/ops/mish.json @@ -0,0 +1,6 @@ +{ + "Mish":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/mod.json b/include/tim/vx/ops/mod.json new file mode 100755 index 0000000..a25a7aa --- /dev/null +++ b/include/tim/vx/ops/mod.json @@ -0,0 +1,12 @@ +{ + "Mod":{ + "parameters": + [ + {"name": "fmod", + "dtype": "int32_t", + "Optional": "true", + "default": "0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/moments.json b/include/tim/vx/ops/moments.json new file mode 100755 index 0000000..a491b04 --- /dev/null +++ b/include/tim/vx/ops/moments.json @@ -0,0 +1,15 @@ +{ + "Moments":{ + "parameters": + [ + {"name": "axis", + "dtype": "std::vector" + }, + {"name": "keep_dims", + "dtype": "bool", + "Optional": "true", + "default": "false" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/mul.json b/include/tim/vx/ops/mul.json new file mode 100755 index 0000000..272fb27 --- /dev/null +++ b/include/tim/vx/ops/mul.json @@ -0,0 +1,12 @@ +{ + "Multiply":{ + "parameters": + [ + {"name": "scale", + "dtype": "float", + "Optional":"true", + "default":"1.0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/nbg.json b/include/tim/vx/ops/nbg.json new file mode 100755 index 0000000..c60fbbb --- /dev/null +++ b/include/tim/vx/ops/nbg.json @@ -0,0 +1,16 @@ +{ + "NBG":{ + "parameters": + [ + {"name":"binary", + "dtype": "char*" + }, + {"name": "input_count", + "dtype": "size_t" + }, + {"name": "output_count", + "dtype": "size_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/neg.json b/include/tim/vx/ops/neg.json new file mode 100755 index 0000000..4327146 --- /dev/null +++ b/include/tim/vx/ops/neg.json @@ -0,0 +1,6 @@ +{ + "Neg":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/notEqual.json b/include/tim/vx/ops/notEqual.json new file mode 100755 index 0000000..fa66841 --- /dev/null +++ b/include/tim/vx/ops/notEqual.json @@ -0,0 +1,6 @@ +{ + "NotEqual":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/onehot.json b/include/tim/vx/ops/onehot.json new file mode 100755 index 0000000..c0fcafe --- /dev/null +++ b/include/tim/vx/ops/onehot.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/pad.json b/include/tim/vx/ops/pad.json new file mode 100755 index 0000000..7c5fc7a --- /dev/null +++ b/include/tim/vx/ops/pad.json @@ -0,0 +1,25 @@ +{ + "Pad":{ + "parameters": + [ + {"name":"front_size", + "dtype": "std::vector" + }, + {"name": "back_size", + "dtype": "std::vector" + }, + {"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"] + } + ] + } +} diff --git a/include/tim/vx/ops/padv2.json b/include/tim/vx/ops/padv2.json new file mode 100755 index 0000000..6465b80 --- /dev/null +++ b/include/tim/vx/ops/padv2.json @@ -0,0 +1,25 @@ +{ + "PadV2":{ + "parameters": + [ + {"name":"front_size", + "dtype": "std::vector" + }, + {"name": "back_size", + "dtype": "std::vector" + }, + {"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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/pool1d.json b/include/tim/vx/ops/pool1d.json new file mode 100755 index 0000000..be29366 --- /dev/null +++ b/include/tim/vx/ops/pool1d.json @@ -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" + }, + {"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"] + } + + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/pool2d.json b/include/tim/vx/ops/pool2d.json new file mode 100755 index 0000000..13096b8 --- /dev/null +++ b/include/tim/vx/ops/pool2d.json @@ -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" + }, + {"name":"ksize", + "dtype":"std::array" + }, + {"name":"stride", + "dtype":"std::array" + }, + {"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"] + } + + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/pow.json b/include/tim/vx/ops/pow.json new file mode 100755 index 0000000..9156ead --- /dev/null +++ b/include/tim/vx/ops/pow.json @@ -0,0 +1,6 @@ +{ + "Pow":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/prelu.json b/include/tim/vx/ops/prelu.json new file mode 100755 index 0000000..d034c3d --- /dev/null +++ b/include/tim/vx/ops/prelu.json @@ -0,0 +1,10 @@ +{ + "Prelu":{ + "parameters": + [ + {"name":"axis", + "dtype": "int" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/rcp.json b/include/tim/vx/ops/rcp.json new file mode 100755 index 0000000..f2e01dd --- /dev/null +++ b/include/tim/vx/ops/rcp.json @@ -0,0 +1,6 @@ +{ + "Rcp":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reduceall.json b/include/tim/vx/ops/reduceall.json new file mode 100755 index 0000000..d4417a4 --- /dev/null +++ b/include/tim/vx/ops/reduceall.json @@ -0,0 +1,13 @@ +{ + "ReduceAll":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + }, + {"name":"keep_dims", + "dtype": "bool" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reduceany.json b/include/tim/vx/ops/reduceany.json new file mode 100755 index 0000000..67fff85 --- /dev/null +++ b/include/tim/vx/ops/reduceany.json @@ -0,0 +1,13 @@ +{ + "ReduceAny":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + }, + {"name":"keep_dims", + "dtype": "bool" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reducemax.json b/include/tim/vx/ops/reducemax.json new file mode 100755 index 0000000..d0e1bc4 --- /dev/null +++ b/include/tim/vx/ops/reducemax.json @@ -0,0 +1,13 @@ +{ + "ReduceMax":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + }, + {"name":"keep_dims", + "dtype": "bool" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reducemean.json b/include/tim/vx/ops/reducemean.json new file mode 100755 index 0000000..51b66fe --- /dev/null +++ b/include/tim/vx/ops/reducemean.json @@ -0,0 +1,13 @@ +{ + "ReduceMean":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + }, + {"name":"keep_dims", + "dtype": "bool" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reducemin.json b/include/tim/vx/ops/reducemin.json new file mode 100755 index 0000000..0e9730e --- /dev/null +++ b/include/tim/vx/ops/reducemin.json @@ -0,0 +1,13 @@ +{ + "ReduceMin":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + }, + {"name":"keep_dims", + "dtype": "bool" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reduceprod.json b/include/tim/vx/ops/reduceprod.json new file mode 100755 index 0000000..d12eefe --- /dev/null +++ b/include/tim/vx/ops/reduceprod.json @@ -0,0 +1,13 @@ +{ + "ReduceProd":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + }, + {"name":"keep_dims", + "dtype": "bool" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reducesum.json b/include/tim/vx/ops/reducesum.json new file mode 100755 index 0000000..1e87502 --- /dev/null +++ b/include/tim/vx/ops/reducesum.json @@ -0,0 +1,13 @@ +{ + "ReduceSum":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + }, + {"name":"keep_dims", + "dtype": "bool" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/relu.json b/include/tim/vx/ops/relu.json new file mode 100755 index 0000000..e4066ac --- /dev/null +++ b/include/tim/vx/ops/relu.json @@ -0,0 +1,6 @@ +{ + "Relu":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/relu1.json b/include/tim/vx/ops/relu1.json new file mode 100755 index 0000000..f6a7b27 --- /dev/null +++ b/include/tim/vx/ops/relu1.json @@ -0,0 +1,6 @@ +{ + "Relu1":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/relu6.json b/include/tim/vx/ops/relu6.json new file mode 100755 index 0000000..b8cd44a --- /dev/null +++ b/include/tim/vx/ops/relu6.json @@ -0,0 +1,6 @@ +{ + "Relu6":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reorg.json b/include/tim/vx/ops/reorg.json new file mode 100755 index 0000000..07b8f80 --- /dev/null +++ b/include/tim/vx/ops/reorg.json @@ -0,0 +1,10 @@ +{ + "Reorg":{ + "parameters": + [ + {"name":"stride", + "dtype": "uint32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/reshape.json b/include/tim/vx/ops/reshape.json new file mode 100755 index 0000000..3be30dc --- /dev/null +++ b/include/tim/vx/ops/reshape.json @@ -0,0 +1,10 @@ +{ + "Reshape":{ + "parameters": + [ + {"name": "size", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/resize.json b/include/tim/vx/ops/resize.json new file mode 100755 index 0000000..22050d0 --- /dev/null +++ b/include/tim/vx/ops/resize.json @@ -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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/resize1d.json b/include/tim/vx/ops/resize1d.json new file mode 100755 index 0000000..3a08313 --- /dev/null +++ b/include/tim/vx/ops/resize1d.json @@ -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"] + } + ] + } +} diff --git a/include/tim/vx/ops/reverse.json b/include/tim/vx/ops/reverse.json new file mode 100755 index 0000000..1f7ab9b --- /dev/null +++ b/include/tim/vx/ops/reverse.json @@ -0,0 +1,10 @@ +{ + "Reverse":{ + "parameters": + [ + {"name":"axis", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/rnncell.json b/include/tim/vx/ops/rnncell.json new file mode 100755 index 0000000..46793a0 --- /dev/null +++ b/include/tim/vx/ops/rnncell.json @@ -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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/roialign.json b/include/tim/vx/ops/roialign.json new file mode 100755 index 0000000..b5c4a64 --- /dev/null +++ b/include/tim/vx/ops/roialign.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/roipool.json b/include/tim/vx/ops/roipool.json new file mode 100755 index 0000000..fbb33ce --- /dev/null +++ b/include/tim/vx/ops/roipool.json @@ -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" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/round.json b/include/tim/vx/ops/round.json new file mode 100755 index 0000000..49836fc --- /dev/null +++ b/include/tim/vx/ops/round.json @@ -0,0 +1,6 @@ +{ + "Round":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/rsqrt.json b/include/tim/vx/ops/rsqrt.json new file mode 100755 index 0000000..f98c193 --- /dev/null +++ b/include/tim/vx/ops/rsqrt.json @@ -0,0 +1,6 @@ +{ + "Rsqrt":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/scatternd.json b/include/tim/vx/ops/scatternd.json new file mode 100755 index 0000000..af81333 --- /dev/null +++ b/include/tim/vx/ops/scatternd.json @@ -0,0 +1,10 @@ +{ + "ScatterND":{ + "parameters": + [ + {"name":"shape", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/select.json b/include/tim/vx/ops/select.json new file mode 100755 index 0000000..53f49ad --- /dev/null +++ b/include/tim/vx/ops/select.json @@ -0,0 +1,6 @@ +{ + "Select":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/selu.json b/include/tim/vx/ops/selu.json new file mode 100755 index 0000000..2876734 --- /dev/null +++ b/include/tim/vx/ops/selu.json @@ -0,0 +1,17 @@ +{ + "Selu":{ + "parameters": + [ + {"name":"alpha", + "dtype": "float", + "Optional":"true", + "default":"1.67326" + }, + {"name":"gamma", + "dtype": "float", + "Optional":"true", + "default":"1.0507" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/shufflechannel.json b/include/tim/vx/ops/shufflechannel.json new file mode 100755 index 0000000..3c4d148 --- /dev/null +++ b/include/tim/vx/ops/shufflechannel.json @@ -0,0 +1,13 @@ +{ + "ShuffleChannel":{ + "parameters": + [ + {"name":"num_groups", + "dtype": "int32_t" + }, + {"name":"index_axis", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/sigmoid.json b/include/tim/vx/ops/sigmoid.json new file mode 100755 index 0000000..0ae9c8c --- /dev/null +++ b/include/tim/vx/ops/sigmoid.json @@ -0,0 +1,6 @@ +{ + "Sigmoid":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/sign.json b/include/tim/vx/ops/sign.json new file mode 100755 index 0000000..1ba1b5a --- /dev/null +++ b/include/tim/vx/ops/sign.json @@ -0,0 +1,6 @@ +{ + "Sign":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/signal_frame.json b/include/tim/vx/ops/signal_frame.json new file mode 100755 index 0000000..92f627b --- /dev/null +++ b/include/tim/vx/ops/signal_frame.json @@ -0,0 +1,23 @@ +{ + "SignalFrame":{ + "parameters": + [ + {"name":"window_length", + "dtype": "uint32_t" + }, + {"name":"step", + "dtype": "uint32_t" + }, + {"name":"pad_end", + "dtype": "uint32_t", + "Optional":"true", + "default":"0" + }, + {"name":"axis", + "dtype": "uint32_t", + "Optional":"true", + "default":"0" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/sin.json b/include/tim/vx/ops/sin.json new file mode 100755 index 0000000..202d0d9 --- /dev/null +++ b/include/tim/vx/ops/sin.json @@ -0,0 +1,6 @@ +{ + "Sin":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/slice.json b/include/tim/vx/ops/slice.json new file mode 100755 index 0000000..fa14ee2 --- /dev/null +++ b/include/tim/vx/ops/slice.json @@ -0,0 +1,16 @@ +{ + "Slice":{ + "parameters": + [ + {"name":"dims", + "dtype": "uint32_t" + }, + {"name":"start", + "dtype": "std::vector" + }, + {"name":"length", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/softmax.json b/include/tim/vx/ops/softmax.json new file mode 100755 index 0000000..850a51b --- /dev/null +++ b/include/tim/vx/ops/softmax.json @@ -0,0 +1,13 @@ +{ + "Softmax":{ + "parameters": + [ + {"name": "beta", + "dtype": "float" + }, + {"name": "axis", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/softrelu.json b/include/tim/vx/ops/softrelu.json new file mode 100755 index 0000000..433689d --- /dev/null +++ b/include/tim/vx/ops/softrelu.json @@ -0,0 +1,6 @@ +{ + "SoftRelu":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/softsign.json b/include/tim/vx/ops/softsign.json new file mode 100755 index 0000000..bbcc93b --- /dev/null +++ b/include/tim/vx/ops/softsign.json @@ -0,0 +1,6 @@ +{ + "SoftSign":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/space2batch.json b/include/tim/vx/ops/space2batch.json new file mode 100755 index 0000000..3ed9451 --- /dev/null +++ b/include/tim/vx/ops/space2batch.json @@ -0,0 +1,21 @@ +{ + "Space2Batch":{ + "parameters": + [ + {"name":"block_size", + "dtype": "std::vector" + }, + {"name": "pad", + "dtype": "std::vector" + }, + {"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"] + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/space2depth.json b/include/tim/vx/ops/space2depth.json new file mode 100755 index 0000000..eecfc0f --- /dev/null +++ b/include/tim/vx/ops/space2depth.json @@ -0,0 +1,18 @@ +{ + "SpaceToDepth":{ + "parameters": + [ + {"name":"block_size", + "dtype": "std::vector" + }, + {"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"] + } + ] + } +} diff --git a/include/tim/vx/ops/spatialtransformer.json b/include/tim/vx/ops/spatialtransformer.json new file mode 100755 index 0000000..e2c3d63 --- /dev/null +++ b/include/tim/vx/ops/spatialtransformer.json @@ -0,0 +1,54 @@ +{ + "SpatialTransformer":{ + "parameters": + [ + {"name":"output_h", + "dtype": "uint32_t" + }, + {"name": "output_w", + "dtype": "uint32_t" + }, + {"name":"has_theta_1_1", + "dtype": "bool" + }, + {"name": "has_theta_1_2", + "dtype": "bool" + }, + {"name":"has_theta_1_3", + "dtype": "bool" + }, + {"name": "has_theta_2_1", + "dtype": "bool" + }, + {"name":"has_theta_2_2", + "dtype": "bool" + }, + {"name": "has_theta_2_3", + "dtype": "bool" + }, + {"name":"theta_1_1", + "dtype": "float" + }, + {"name": "theta_1_2", + "dtype": "float" + }, + {"name":"theta_1_3", + "dtype": "float" + }, + {"name": "theta_2_1", + "dtype": "float" + }, + {"name":"theta_2_2", + "dtype": "float" + }, + {"name": "theta_2_3", + "dtype": "float" + }, + {"name":"align_corners", + "dtype": "bool", + "Optional":"true", + "default":"false" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/split.json b/include/tim/vx/ops/split.json new file mode 100755 index 0000000..fd093da --- /dev/null +++ b/include/tim/vx/ops/split.json @@ -0,0 +1,13 @@ +{ + "Split":{ + "parameters": + [ + {"name":"axis", + "dtype": "uint32_t" + }, + {"name": "slices", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/sqrt.json b/include/tim/vx/ops/sqrt.json new file mode 100755 index 0000000..7fbee85 --- /dev/null +++ b/include/tim/vx/ops/sqrt.json @@ -0,0 +1,6 @@ +{ + "Sqrt":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/square.json b/include/tim/vx/ops/square.json new file mode 100755 index 0000000..21b5b03 --- /dev/null +++ b/include/tim/vx/ops/square.json @@ -0,0 +1,6 @@ +{ + "Square":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/squeeze.json b/include/tim/vx/ops/squeeze.json new file mode 100755 index 0000000..84975b5 --- /dev/null +++ b/include/tim/vx/ops/squeeze.json @@ -0,0 +1,10 @@ +{ + "Squeeze":{ + "parameters": + [ + {"name": "axis", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/stack.json b/include/tim/vx/ops/stack.json new file mode 100755 index 0000000..27f399c --- /dev/null +++ b/include/tim/vx/ops/stack.json @@ -0,0 +1,13 @@ +{ + "Stack":{ + "parameters": + [ + {"name":"axis", + "dtype": "uint32_t" + }, + {"name": "input_cnt", + "dtype": "int" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/stridedslice.json b/include/tim/vx/ops/stridedslice.json new file mode 100755 index 0000000..e48ea70 --- /dev/null +++ b/include/tim/vx/ops/stridedslice.json @@ -0,0 +1,25 @@ +{ + "StridedSlice":{ + "parameters": + [ + {"name":"begin_dims", + "dtype": "std::vector" + }, + {"name": "end_dims", + "dtype": "std::vector" + }, + {"name":"stride_dims", + "dtype": "std::vector" + }, + {"name": "begin_mask", + "dtype": "int32_t" + }, + {"name":"end_mask", + "dtype": "int32_t" + }, + {"name": "shrink_axis_mask", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/sub.json b/include/tim/vx/ops/sub.json new file mode 100755 index 0000000..916989c --- /dev/null +++ b/include/tim/vx/ops/sub.json @@ -0,0 +1,6 @@ +{ + "Sub":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/svdf.json b/include/tim/vx/ops/svdf.json new file mode 100755 index 0000000..779b4b2 --- /dev/null +++ b/include/tim/vx/ops/svdf.json @@ -0,0 +1,16 @@ +{ + "Svdf":{ + "parameters": + [ + {"name":"rank", + "dtype": "int32_t" + }, + {"name":"num_units", + "dtype": "int32_t" + }, + {"name":"spectrogram_length", + "dtype": "int32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/swish.json b/include/tim/vx/ops/swish.json new file mode 100755 index 0000000..041529a --- /dev/null +++ b/include/tim/vx/ops/swish.json @@ -0,0 +1,6 @@ +{ + "Swish":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/tanh.json b/include/tim/vx/ops/tanh.json new file mode 100755 index 0000000..81fe8b4 --- /dev/null +++ b/include/tim/vx/ops/tanh.json @@ -0,0 +1,6 @@ +{ + "Tanh":{ + "parameters": + [] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/tile.json b/include/tim/vx/ops/tile.json new file mode 100755 index 0000000..46aecb0 --- /dev/null +++ b/include/tim/vx/ops/tile.json @@ -0,0 +1,10 @@ +{ + "Tile":{ + "parameters": + [ + {"name":"multiples", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/topk.json b/include/tim/vx/ops/topk.json new file mode 100755 index 0000000..9b60d67 --- /dev/null +++ b/include/tim/vx/ops/topk.json @@ -0,0 +1,10 @@ +{ + "Topk":{ + "parameters": + [ + {"name":"k", + "dtype": "uint32_t" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/transpose.json b/include/tim/vx/ops/transpose.json new file mode 100755 index 0000000..777b050 --- /dev/null +++ b/include/tim/vx/ops/transpose.json @@ -0,0 +1,10 @@ +{ + "Transpose":{ + "parameters": + [ + {"name":"perm", + "dtype": "std::vector" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/unidirectionalsequenceGRU.json b/include/tim/vx/ops/unidirectionalsequenceGRU.json new file mode 100755 index 0000000..aa780d7 --- /dev/null +++ b/include/tim/vx/ops/unidirectionalsequenceGRU.json @@ -0,0 +1,47 @@ +{ + "UnidirectionalSequenceGRU":{ + "parameters": + [ + {"name":"num_units", + "dtype": "uint32_t" + }, + {"name": "activation", + "dtype": "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType", + "Optional":"true", + "default":"tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kTANH", + "range": ["tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kNONE", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kRELU", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kRELU6", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kTANH", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kSIGMOID", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kHARDSIGMOID"] + }, + {"name": "recurrent_activation", + "dtype": "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType", + "Optional":"true", + "default":"tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kSIGMOID", + "range": ["tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kNONE", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kRELU", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kRELU6", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kTANH", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kSIGMOID", + "tim::vx::ops::UnidirectionalSequenceGRU::ActivationType::kHARDSIGMOID"] + }, + {"name":"reset_after", + "dtype": "vsi_bool", + "Optional":"true", + "default":"TRUE" + }, + {"name":"return_sequences", + "dtype": "vsi_bool", + "Optional":"true", + "default":"FALSE" + }, + {"name":"time_major", + "dtype": "vsi_bool", + "Optional":"true", + "default":"TRUE" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/unidirectionalsequenceLstm.json b/include/tim/vx/ops/unidirectionalsequenceLstm.json new file mode 100755 index 0000000..7b0329a --- /dev/null +++ b/include/tim/vx/ops/unidirectionalsequenceLstm.json @@ -0,0 +1,46 @@ +{ + "UnidirectionalSequenceLstm":{ + "parameters": + [ + {"name":"cell_clip", + "dtype": "float" + }, + {"name":"proj_clip", + "dtype": "float" + }, + {"name": "act_type", + "dtype": "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType", + "range": ["tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kNONE", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kRELU", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kRELU6", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kTANH", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kSIGMOID", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kHARDSIGMOID"] + }, + {"name":"forget_bias", + "dtype": "float" + }, + {"name":"time_major", + "dtype": "bool", + "Optional":"true", + "default":"false" + }, + {"name": "recurrent_act_type", + "dtype": "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType", + "Optional":"true", + "default":"tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kSIGMOID", + "range": ["tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kNONE", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kRELU", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kRELU6", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kTANH", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kSIGMOID", + "tim::vx::ops::UnidirectionalSequenceLstm::ActivationType::kHARDSIGMOID"] + }, + {"name":"return_sequence", + "dtype": "bool", + "Optional":"true", + "default":"flase" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/unidirectionalsequenceRNN.json b/include/tim/vx/ops/unidirectionalsequenceRNN.json new file mode 100755 index 0000000..04d2693 --- /dev/null +++ b/include/tim/vx/ops/unidirectionalsequenceRNN.json @@ -0,0 +1,22 @@ +{ + "UnidirectionalSequenceRnn":{ + "parameters": + [ + {"name": "act_type", + "dtype": "tim::vx::ops::UnidirectionalSequenceRnn::ActivationType", + "range": ["tim::vx::ops::UnidirectionalSequenceRnn::ActivationType::kNONE", + "tim::vx::ops::UnidirectionalSequenceRnn::ActivationType::kRELU", + "tim::vx::ops::UnidirectionalSequenceRnn::ActivationType::kRELU1", + "tim::vx::ops::UnidirectionalSequenceRnn::ActivationType::kRELU6", + "tim::vx::ops::UnidirectionalSequenceRnn::ActivationType::kTANH", + "tim::vx::ops::UnidirectionalSequenceRnn::ActivationType::kSIGMOID", + "tim::vx::ops::UnidirectionalSequenceRnn::ActivationType::kHARDSIGMOID"] + }, + {"name":"time_major", + "dtype": "bool", + "Optional":"true", + "default":"false" + } + ] + } +} \ No newline at end of file diff --git a/include/tim/vx/ops/unstack.json b/include/tim/vx/ops/unstack.json new file mode 100755 index 0000000..d2d8128 --- /dev/null +++ b/include/tim/vx/ops/unstack.json @@ -0,0 +1,13 @@ +{ + "Unstack":{ + "parameters": + [ + {"name":"axis", + "dtype": "int32_t" + }, + {"name":"output_num", + "dtype": "uint32_t" + } + ] + } +} \ No newline at end of file