Home
last modified time | relevance | path

Searched refs:recovered_cell_weights (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc420 TfLiteTensor* recovered_cell_weights = in Prepare() local
422 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
423 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
425 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
430 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
530 TfLiteTensor* recovered_cell_weights = in Eval() local
550 prod_scaling_factors, recovered_cell_weights, input_quantized, in Eval()
Dbidirectional_sequence_lstm.cc713 TfLiteTensor* recovered_cell_weights = in Prepare() local
715 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
716 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
718 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
723 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
969 TfLiteTensor* recovered_cell_weights = in Eval() local
993 recovered_cell_weights, input_quantized, aux_input_quantized, in Eval()
1016 recovered_cell_weights, input_quantized, aux_input_quantized, in Eval()
Dlstm.cc488 TfLiteTensor* recovered_cell_weights = in Prepare() local
490 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
491 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
493 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
498 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
609 TfLiteTensor* recovered_cell_weights = in Eval() local
627 scaling_factors, prod_scaling_factors, recovered_cell_weights, in Eval()
Dlstm_eval.cc471 float* recovered_cell_weights, int8_t* quantized_input_ptr_batch, in LstmStepWithAuxInput()
661 recovered_cell_weights); in LstmStepWithAuxInput()
663 recovered_cell_weights, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput()
684 recovered_cell_weights); in LstmStepWithAuxInput()
686 recovered_cell_weights, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput()
736 recovered_cell_weights); in LstmStepWithAuxInput()
738 recovered_cell_weights, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput()
1095 TfLiteTensor* recovered_cell_weights, TfLiteTensor* input_quantized, in EvalHybrid()
1235 float* recovered_cell_weights_ptr = recovered_cell_weights->data.f; in EvalHybrid()
Dlstm_eval.h82 TfLiteTensor* recovered_cell_weights, TfLiteTensor* input_quantized,