Searched refs:batchInputShape (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 440 Shape batchInputShape = input_shape; in LSTMEvalFloat32() local
441 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() local
561 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat16()
722 LSTMStep(params, inputCurrentTimeStep, batchInputShape, in LSTMEvalFloat16()
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 129 TensorShape batchInputShape = TensorShape({batchSize, inputSize}); in Execute() local
130 lstmInputInfo.SetShape(batchInputShape); in Execute()
|