Searched refs:bw_input_to_output_weights (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm.cc | 450 const TfLiteTensor* bw_input_to_output_weights; in Prepare() local
453 &bw_input_to_output_weights)); in Prepare()
454 const int n_bw_cell = bw_input_to_output_weights->dims->data[0]; in Prepare()
455 TF_LITE_ENSURE_EQ(context, bw_input_to_output_weights->dims->size, 2); in Prepare()
456 TF_LITE_ENSURE_EQ(context, bw_input_to_output_weights->dims->data[1], in Prepare()
458 TF_LITE_ENSURE_EQ(context, bw_input_to_output_weights->type, in Prepare()
1055 const TfLiteTensor* bw_input_to_output_weights; in Eval() local
1058 &bw_input_to_output_weights)); in Eval()
1197 bw_input_to_cell_weights, bw_input_to_output_weights, in Eval()
1307 /*input_to_cell_weights_ledger*/ nullptr, bw_input_to_output_weights, in Eval()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4086 TFL_OperandHasRank<21, 2>, // bw_input_to_output_weights
4146 TFL_TensorOf<[F32, I8]>:$bw_input_to_output_weights,
|