Searched refs:batchInputSize (Results 1 – 1 of 1) sorted by relevance
399 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat32() local409 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()412 transposedAuxInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()431 inputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1)); in LSTMEvalFloat32()433 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()437 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat32()519 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat16() local522 std::vector<float> input_float32(maxTime * batchInputSize); in LSTMEvalFloat16()562 std::vector<float> aux_input_float32(maxTime * batchInputSize); in LSTMEvalFloat16()644 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat16()[all …]