22 lines
1.1 KiB
JSON
22 lines
1.1 KiB
JSON
|
{
|
||
|
|
"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"
|
||
|
|
}
|
||
|
|
]
|
||
|
|
}
|
||
|
|
}
|