Searched defs:output_batch_leading_dim (Results 1 – 5 of 5) sorted by relevance
27 int output_batch_leading_dim, in RnnBatchStep()43 int batch_size, int output_batch_leading_dim, in RnnBatchStep()123 int num_units, int batch_size, int output_batch_leading_dim, in RnnBatchStep()149 int batch_size, int output_batch_leading_dim, in RnnBatchStep()
209 const int output_batch_leading_dim = in EvalFloat() local242 const int output_batch_leading_dim = in EvalHybrid() local
851 inline void LstmStepFloat( in LstmStepFloat()1033 inline void LstmStepHybrid( in LstmStepHybrid()1627 inline void LstmStepInteger8x8_8( in LstmStepInteger8x8_8()1814 const int output_batch_leading_dim = in EvalFloat() local2013 const int output_batch_leading_dim = in EvalHybrid() local2258 const int output_batch_leading_dim = in EvalInteger8x8_16() local2496 const int output_batch_leading_dim = in EvalInteger8x8_8() local
40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()408 const int output_batch_leading_dim = in EvalFloat() local
158 inline void LstmStepCalibration( in LstmStepCalibration()331 const int output_batch_leading_dim = in EvalCalibration() local