Home
last modified time | relevance | path

Searched refs:lstm_gate_concat (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNELSTMLayer.cpp503 …TensorInfo lstm_gate_concat = TensorInfo(lstm_weights_concat_shape, 1, input->data_type(… in validate() local
504 …ARM_COMPUTE_RETURN_ON_ERROR(NEConcatenateLayer::validate(lstm_weights, &lstm_gate_concat, Window::… in validate()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLLSTMLayer.cpp529 …TensorInfo lstm_gate_concat = TensorInfo(lstm_weights_concat_shape, 1, input->data_type(… in validate() local
530 …ARM_COMPUTE_RETURN_ON_ERROR(CLConcatenateLayer::validate(lstm_weights, &lstm_gate_concat, Window::… in validate()