Home
last modified time | relevance | path

Searched refs:has_aux_input (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm.cc523 const bool has_aux_input = (fw_aux_input_to_forget_weights != nullptr); in Prepare() local
525 if (has_aux_input) { in Prepare()
564 has_aux_input ? kNumTemporaryTensors : kNumTemporaryTensors - 1); in Prepare()
580 if (has_aux_input && !fw_use_cifg) { in Prepare()
642 if (has_aux_input && !bw_use_cifg) { in Prepare()
836 if (has_aux_input) { in Prepare()
896 if (has_aux_input) { in Prepare()
921 if (has_aux_input) { in Prepare()
947 if (has_aux_input) { in Prepare()
Dbidirectional_sequence_rnn.cc141 const bool has_aux_input = (fw_aux_input_weights != nullptr); in Prepare() local
174 if (has_aux_input) { in Prepare()
193 if (has_aux_input) { in Prepare()
294 const int num_row_sums = has_aux_input ? 3 : 2; in Prepare()
327 if (has_aux_input) { in Prepare()