Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm.cc505 const TfLiteTensor* bw_aux_input_to_output_weights = in Prepare() local
514 (bw_aux_input_to_output_weights != nullptr)) || in Prepare()
520 (bw_aux_input_to_output_weights == nullptr)); in Prepare()
838 n_cell = std::max(n_cell, bw_aux_input_to_output_weights->dims->data[0]); in Prepare()
1137 const TfLiteTensor* bw_aux_input_to_output_weights = in Eval() local
1207 bw_aux_input_to_cell_weights, bw_aux_input_to_output_weights, in Eval()
1324 bw_aux_input_to_cell_weights, bw_aux_input_to_output_weights, in Eval()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td4189 TFL_TensorOfOrNone<[F32, I8]>:$bw_aux_input_to_output_weights,