Searched refs:cell_to_forget_weights_ptr (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_eval.cc | 107 const float* cell_to_forget_weights_ptr, in LstmStepWithAuxInput() argument 229 cell_to_forget_weights_ptr, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput() 455 const int8_t* cell_to_forget_weights_ptr, in LstmStepWithAuxInput() 682 tensor_utils::VectorScalarMultiply(cell_to_forget_weights_ptr, n_cell, in LstmStepWithAuxInput() 946 const float* cell_to_forget_weights_ptr = in EvalFloat() local 1004 cell_to_forget_weights_ptr, cell_to_output_weights_ptr, in EvalFloat() 1050 cell_to_forget_weights_ptr, cell_to_output_weights_ptr, in EvalFloat() 1156 int8_t* cell_to_forget_weights_ptr = nullptr; in EvalHybrid() local 1167 cell_to_forget_weights_ptr = in EvalHybrid() 1301 cell_to_forget_weights_ptr, cell_to_forget_weights_scale, in EvalHybrid() [all …]
|