Searched refs:batchInputShape (Results 1 – 1 of 1) sorted by relevance
440 Shape batchInputShape = input_shape; in LSTMEvalFloat32() local441 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat32()484 LSTMStep(params, inputCurrentTimeStep, batchInputShape, input_to_input_weights_buffer, in LSTMEvalFloat32()560 Shape batchInputShape = input_shape; in LSTMEvalFloat16() local561 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat16()722 LSTMStep(params, inputCurrentTimeStep, batchInputShape, in LSTMEvalFloat16()