@@ -108,7 +108,7 @@ one step of kernel (M or L*M elements for stacked weights matrix).
108108For the other modes (one-to-one or batch-to-batch), kernel does not
109109use the intermediate result tensor and this field might not be
110110initialized. For more information about configuration structure, see
111- :ref: `fn_conf_brnn `.
111+ :ref: `fn_conf_lstm `.
112112
113113.. caution ::
114114 Ensure that you allocate memory for all tensors (including
@@ -222,59 +222,61 @@ Parameters
222222''''''''''
223223
224224.. table :: Kernel Interface Parameters
225-
226- +-----------------------+-----------------------+
227- | **Parameters ** | **Description ** |
228- +=======================+=======================+
229- | | |
230- | ``in `` | [IN] Pointer to input |
231- | | tensor |
232- +-----------------------+-----------------------+
233- | | |
234- | ``prev_out `` | [IN] Pointer to |
235- | | previous output |
236- | | tensor |
237- +-----------------------+-----------------------+
238- | | |
239- | ``weights `` | [IN] Pointer to |
240- | | weights tensor |
241- +-----------------------+-----------------------+
242- | | |
243- | ``bias `` | [IN] Pointer to |
244- | | biases tensor |
245- +-----------------------+-----------------------+
246- | | |
247- | ``cfg `` | [IN/OUT] Pointer to |
248- | | configuration |
249- | | structure |
250- +-----------------------+-----------------------+
251- | | |
252- | ``out `` | [OUT] Pointer to |
253- | | output tensor. Result |
254- | | is stored here |
255- +-----------------------+-----------------------+
225+ :widths: 20,130
226+
227+ +-----------------------+-----------------------+
228+ | **Parameters ** | **Description ** |
229+ +=======================+=======================+
230+ | | |
231+ | ``in `` | [IN] Pointer to input |
232+ | | tensor |
233+ +-----------------------+-----------------------+
234+ | | |
235+ | ``prev_out `` | [IN] Pointer to |
236+ | | previous output |
237+ | | tensor |
238+ +-----------------------+-----------------------+
239+ | | |
240+ | ``weights `` | [IN] Pointer to |
241+ | | weights tensor |
242+ +-----------------------+-----------------------+
243+ | | |
244+ | ``bias `` | [IN] Pointer to |
245+ | | biases tensor |
246+ +-----------------------+-----------------------+
247+ | | |
248+ | ``cfg `` | [IN/OUT] Pointer to |
249+ | | configuration |
250+ | | structure |
251+ +-----------------------+-----------------------+
252+ | | |
253+ | ``out `` | [OUT] Pointer to |
254+ | | output tensor. Result |
255+ | | is stored here |
256+ +-----------------------+-----------------------+
256257
257258.. _kernel-specializations-2 :
258259
259260Kernel Specializations
260261^^^^^^^^^^^^^^^^^^^^^^
261262
262263.. table :: Non-Specialized Functions
263-
264- +---------------------------------------+-----------------------------------+
265- | **Function ** | **Description ** |
266- +=======================================+===================================+
267- | ``mli_krn_basic_rnn_cell_fx8 `` | General function; 8bit FX |
268- | | elements; |
269- +---------------------------------------+-----------------------------------+
270- | ``mli_krn_basic_rnn_cell_fx16 `` | General function; 16bit FX |
271- | | elements; |
272- +---------------------------------------+-----------------------------------+
273- | ``mli_krn_basic_rnn_cell_fx8w16d `` | General function; FX tensors |
274- | | (8bit weights and biases, 16 bit |
275- | | input, state, cell, output and |
276- | | intermediate data); |
277- +---------------------------------------+-----------------------------------+
264+ :widths: 20,130
265+
266+ +---------------------------------------+-----------------------------------+
267+ | **Function ** | **Description ** |
268+ +=======================================+===================================+
269+ | ``mli_krn_basic_rnn_cell_fx8 `` | General function; 8bit FX |
270+ | | elements; |
271+ +---------------------------------------+-----------------------------------+
272+ | ``mli_krn_basic_rnn_cell_fx16 `` | General function; 16bit FX |
273+ | | elements; |
274+ +---------------------------------------+-----------------------------------+
275+ | ``mli_krn_basic_rnn_cell_fx8w16d `` | General function; FX tensors |
276+ | | (8bit weights and biases, 16 bit |
277+ | | input, state, cell, output and |
278+ | | intermediate data); |
279+ +---------------------------------------+-----------------------------------+
278280
279281.. _conditions-for-applying-the-kernel-2 :
280282
0 commit comments