Searched defs:LSTMModel1 (Results 1 – 1 of 1) sorted by relevance
25 struct LSTMModel1 { struct26 const std::vector<int32_t> input_shape = {5, 2, 10};27 const std::vector<int32_t> wih_shape = {8, 10};28 const std::vector<int32_t> whh_shape = {8, 2};29 const std::vector<int32_t> bias_shape = {16};30 const std::vector<int32_t> hx_shape = {2, 2};31 const std::vector<int32_t> output_shape = {5, 2};33 float inputValue[5][2][10] = {1};34 float wihValue[8][10] = {1};35 float whhValue[8][2] = {1};[all …]