Searched refs:state_num_dims (Results 1 – 2 of 2) sorted by relevance
254 int state_num_dims, Model* model);
1466 int state_num_dims, Model* model) { in CreateOrCheckRnnStateArray() argument1469 if (state_num_dims > 0) { in CreateOrCheckRnnStateArray()1470 num_dims = state_num_dims; in CreateOrCheckRnnStateArray()