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()
208 const int output_batch_leading_dim = in EvalFloat() local241 const int output_batch_leading_dim = in EvalHybrid() local
808 inline void LstmStepFloat( in LstmStepFloat()977 inline void LstmStepHybrid( in LstmStepHybrid()1569 inline void LstmStepInteger8x8_8( in LstmStepInteger8x8_8()1752 const int output_batch_leading_dim = in EvalFloat() local1949 const int output_batch_leading_dim = in EvalHybrid() local2194 const int output_batch_leading_dim = in EvalInteger8x8_16() local2432 const int output_batch_leading_dim = in EvalInteger8x8_8() local
40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()399 const int output_batch_leading_dim = in EvalFloat() local
157 inline void LstmStepCalibration( in LstmStepCalibration()330 const int output_batch_leading_dim = in EvalCalibration() local