Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm.cc441 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()
668 input_quantized->type = fw_input_to_output_weights->type; in Prepare()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td4069 TFL_OperandHasRank<4, 2>, // fw_input_to_output_weights
4116 TFL_TensorOf<[F32, I8]>:$fw_input_to_output_weights,