Searched refs:batchInputSize (Results 1 – 1 of 1) sorted by relevance
447 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat32() local457 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()460 transposedAuxInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()479 inputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1)); in LSTMEvalFloat32()481 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()485 const int batchInputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(batchInputSize); in LSTMEvalFloat32()567 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat16() local570 std::vector<float> input_float32(maxTime * batchInputSize); in LSTMEvalFloat16()610 std::vector<float> aux_input_float32(maxTime * batchInputSize); in LSTMEvalFloat16()692 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat16()[all …]