Searched refs:batchInputSize (Results 1 – 1 of 1) sorted by relevance
442 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat32() local452 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()455 transposedAuxInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()474 inputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1)); in LSTMEvalFloat32()476 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()480 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat32()562 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat16() local565 std::vector<float> input_float32(maxTime * batchInputSize); in LSTMEvalFloat16()605 std::vector<float> aux_input_float32(maxTime * batchInputSize); in LSTMEvalFloat16()687 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat16()[all …]