Home
last modified time | relevance | path

Searched defs:output_batch_leading_dim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc25 int output_batch_leading_dim, in RnnBatchStep()
41 int batch_size, int output_batch_leading_dim, in RnnBatchStep()
124 int num_units, int batch_size, int output_batch_leading_dim, in RnnBatchStep()
146 int batch_size, int output_batch_leading_dim, in RnnBatchStep()
/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc154 const int output_batch_leading_dim = in EvalFloat() local
186 const int output_batch_leading_dim = in EvalHybrid() local
Dlstm_eval.cc93 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
430 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
978 const int output_batch_leading_dim = in EvalFloat() local
1269 const int output_batch_leading_dim = in EvalHybrid() local