Searched refs:numBatches (Results 1 – 3 of 3) 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()
3746 * and shape [numBatches, inputSize] specifying the input to the LSTM3818 * and shape [numBatches, outputSize] specifying the cell state from the3831 * and shape [numBatches, outputSize] which contains a cell state from
3972 * and shape [numBatches, inputSize] specifying the input to the LSTM4044 * and shape [numBatches, outputSize] specifying the cell state from the4057 * and shape [numBatches, outputSize] which contains a cell state from