Searched refs:batchInputShape (Results 1 – 1 of 1) sorted by relevance
397 Shape batchInputShape = input_shape; in LSTMEvalFloat32() local398 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat32()441 LSTMStep(params, inputCurrentTimeStep, batchInputShape, input_to_input_weights_buffer, in LSTMEvalFloat32()517 Shape batchInputShape = input_shape; in LSTMEvalFloat16() local518 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat16()679 LSTMStep(params, inputCurrentTimeStep, batchInputShape, in LSTMEvalFloat16()