Searched refs:state_num_dims (Results 1 – 2 of 2) sorted by relevance
252 int state_num_dims, Model* model);
1485 int state_num_dims, Model* model) { in CreateOrCheckRnnStateArray() argument1488 if (state_num_dims > 0) { in CreateOrCheckRnnStateArray()1489 num_dims = state_num_dims; in CreateOrCheckRnnStateArray()