Searched defs:aux_input_size (Results 1 – 4 of 4) sorted by relevance
40 int input_size, int aux_input_size, int num_units, in RnnBatchStep()145 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
301 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalFloat() local435 const int aux_input_size = (aux_input) ? aux_input->dims->data[2] : 0; in EvalHybrid() local
908 const int aux_input_size = in EvalFloat() local1112 const int aux_input_size = in EvalHybrid() local
674 int aux_input_size = 0; in BidirectionalRNNOpModel() local