Home
last modified time | relevance | path

Searched defs:n_hidden (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dneon_tensor_utils.h142 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
H A Dsse_tensor_utils.h156 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dportable_tensor_utils.h181 int32_t n_hidden, int32_t n_output, in MatrixBatchVectorMultiply()
H A Dportable_tensor_utils.cc412 const int32_t* gate_bias, int32_t n_batch, int32_t n_hidden, in PortableMatrixBatchVectorMultiply()