Lines Matching refs:num_units
66 const uint32_t num_units = SizeOfDimension(input_weights, 0); in Prepare() local
76 hiddenStateShape->dimensions = { batch_size, num_units }; in Prepare()
80 outputShape->dimensions = { batch_size, num_units }; in Prepare()
150 const uint32_t num_units = weightsShape.dimensions[0]; in RNNStep() local
167 const T* hidden_state_in_ptr_batch = hiddenStateInputData + b * num_units; in RNNStep()
183 for (uint32_t o = 0; o < num_units; o++) { in RNNStep()
188 for (uint32_t o = 0; o < num_units; o++) { in RNNStep()
197 for (uint32_t o = 0; o < num_units; o++) { in RNNStep()
206 for (uint32_t o = 0; o < num_units; o++) { in RNNStep()
207 for (uint32_t h = 0; h < num_units; h++) { in RNNStep()
214 for (uint32_t o = 0; o < num_units; o++) { in RNNStep()