| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_lstm.cc | 441 const TfLiteTensor* fw_input_to_output_weights; in Prepare() local
444 &fw_input_to_output_weights)); in Prepare()
445 const int n_fw_cell = fw_input_to_output_weights->dims->data[0]; in Prepare()
446 TF_LITE_ENSURE_EQ(context, fw_input_to_output_weights->dims->size, 2); in Prepare()
447 TF_LITE_ENSURE_EQ(context, fw_input_to_output_weights->dims->data[1], in Prepare()
459 fw_input_to_output_weights->type); in Prepare()
469 fw_input_to_output_weights->type); in Prepare()
480 fw_input_to_output_weights->type); in Prepare()
559 const bool is_hybrid_op = IsHybridOp(input, fw_input_to_output_weights); in Prepare()
672 input_quantized->type = fw_input_to_output_weights->type; in Prepare()
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py | 35 fw_input_to_output_weights = Input( variable
222 fw_input_to_output_weights,
282 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_float16_batch_major.mod.py | 34 fw_input_to_output_weights = Input( variable
222 fw_input_to_output_weights,
282 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_cifg_peephole.mod.py | 34 fw_input_to_output_weights = Input( variable
222 fw_input_to_output_weights,
282 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm.mod.py | 34 fw_input_to_output_weights = Input( variable
222 fw_input_to_output_weights,
282 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_merge_outputs.mod.py | 35 fw_input_to_output_weights = Input( variable
221 fw_input_to_output_weights,
281 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_norm_fw_output.mod.py | 35 fw_input_to_output_weights = Input( variable
223 fw_input_to_output_weights,
283 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_aux_input.mod.py | 36 fw_input_to_output_weights = Input( variable
224 fw_input_to_output_weights,
284 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py | 37 fw_input_to_output_weights = Input( variable
225 fw_input_to_output_weights,
285 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm.mod.py | 98 fw_input_to_output_weights = Input("fw_input_to_output_weights",
250 fw_input_to_output_weights,
315 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| D | bidirectional_sequence_lstm_state_output.mod.py | 34 fw_input_to_output_weights = Input( variable
238 fw_input_to_output_weights,
304 fw_input_to_output_weights: fw_input_to_output_weights_data,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.td | 4470 TFL_OperandHasRank<4, 2>, // fw_input_to_output_weights
4519 TFL_TensorOf<[F32, I8]>:$fw_input_to_output_weights,
|