| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A 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()
1007 const TfLiteTensor* fw_recurrent_to_output_weights; in Eval() local
1010 &fw_recurrent_to_output_weights)); in Eval()
1153 params->merge_outputs ? fw_recurrent_to_output_weights->dims->data[1] : 0; in Eval()
1183 fw_recurrent_to_cell_weights, fw_recurrent_to_output_weights, in Eval()
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py | 44 fw_recurrent_to_output_weights = Input( variable
226 fw_recurrent_to_output_weights,
286 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_float16_batch_major.mod.py | 43 fw_recurrent_to_output_weights = Input( variable
226 fw_recurrent_to_output_weights,
286 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_cifg_peephole.mod.py | 43 fw_recurrent_to_output_weights = Input( variable
226 fw_recurrent_to_output_weights,
286 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm.mod.py | 43 fw_recurrent_to_output_weights = Input( variable
226 fw_recurrent_to_output_weights,
286 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_merge_outputs.mod.py | 44 fw_recurrent_to_output_weights = Input( variable
225 fw_recurrent_to_output_weights,
285 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_norm_fw_output.mod.py | 44 fw_recurrent_to_output_weights = Input( variable
227 fw_recurrent_to_output_weights,
287 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_aux_input.mod.py | 45 fw_recurrent_to_output_weights = Input( variable
228 fw_recurrent_to_output_weights,
288 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py | 46 fw_recurrent_to_output_weights = Input( variable
229 fw_recurrent_to_output_weights,
289 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm.mod.py | 111 fw_recurrent_to_output_weights = Input("fw_recurrent_to_output_weights",
254 fw_recurrent_to_output_weights,
319 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_state_output.mod.py | 43 fw_recurrent_to_output_weights = Input( variable
242 fw_recurrent_to_output_weights,
308 fw_recurrent_to_output_weights: fw_recurrent_to_output_weights_data,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.td | 4474 TFL_OperandHasRank<8, 2>, // fw_recurrent_to_output_weights
4525 TFL_TensorOf<[F32, I8]>:$fw_recurrent_to_output_weights,
|