colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
TIM-VX/include/tim/vx/ops/unidirectionalsequenceGRU.json

47 lines
2.4 KiB
JSON
Raw Normal View History

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>
2023-06-02 07:51:10 +08:00
{
"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"
}
]
}
}