Searched refs:numBatches (Results 1 – 2 of 2) sorted by relevance
433 int numBatches = mInflightBatches.size(); in getBatch() local434 if (numBatches == 0) { in getBatch()438 uint32_t frameMax = mInflightBatches[numBatches - 1]->mLastFrame; in getBatch()442 for (int i = 0; i < numBatches; i++) { in getBatch()
3769 * and shape [numBatches, inputSize] specifying the input to the LSTM3841 * and shape [numBatches, outputSize] specifying the cell state from the3854 * and shape [numBatches, outputSize] which contains a cell state from