Searched refs:fw_recurrent_to_output_weights (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm.cc | 461 const TfLiteTensor* fw_recurrent_to_output_weights; in Prepare() local
464 &fw_recurrent_to_output_weights)); in Prepare()
465 TF_LITE_ENSURE_EQ(context, fw_recurrent_to_output_weights->dims->size, 2); in Prepare()
466 TF_LITE_ENSURE_EQ(context, fw_recurrent_to_output_weights->dims->data[0], in Prepare()
468 TF_LITE_ENSURE_EQ(context, fw_recurrent_to_output_weights->type, in Prepare()
470 const int n_fw_output = fw_recurrent_to_output_weights->dims->data[1]; in Prepare()
1003 const TfLiteTensor* fw_recurrent_to_output_weights; in Eval() local
1006 &fw_recurrent_to_output_weights)); in Eval()
1149 params->merge_outputs ? fw_recurrent_to_output_weights->dims->data[1] : 0; in Eval()
1179 fw_recurrent_to_cell_weights, fw_recurrent_to_output_weights, in Eval()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4073 TFL_OperandHasRank<8, 2>, // fw_recurrent_to_output_weights
4122 TFL_TensorOf<[F32, I8]>:$fw_recurrent_to_output_weights,
|