Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm.cc810 TfLiteTensor* recovered_cell_weights; in Prepare() local
813 &recovered_cell_weights)); in Prepare()
814 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
815 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
817 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
822 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
1243 TfLiteTensor* recovered_cell_weights; in Eval() local
1246 &recovered_cell_weights)); in Eval()
1292 prod_scaling_factors, recovered_cell_weights, input_quantized, in Eval()
1333 prod_scaling_factors, recovered_cell_weights, input_quantized, in Eval()
Dlstm_eval.cc395 float* recovered_cell_weights = scratch1; in CalculateLstmGateHybrid() local
398 recovered_cell_weights); in CalculateLstmGateHybrid()
400 recovered_cell_weights, n_cell, cell_state, n_batch, gate); in CalculateLstmGateHybrid()
1028 float* scaling_factors_scratch, float* recovered_cell_weights, in LstmStepHybrid()
1153 scaling_factors_scratch, recovered_cell_weights, accum_scratch_ptr); in LstmStepHybrid()
1170 scaling_factors_scratch, recovered_cell_weights, accum_scratch_ptr); in LstmStepHybrid()
1186 context, scaling_factors_scratch, recovered_cell_weights, in LstmStepHybrid()
1207 scaling_factors_scratch, recovered_cell_weights, accum_scratch_ptr); in LstmStepHybrid()
1905 TfLiteTensor* prod_scaling_factors, TfLiteTensor* recovered_cell_weights, in EvalHybrid()
2035 GetTensorData<float>(recovered_cell_weights), in EvalHybrid()
[all …]
Dunidirectional_sequence_lstm.cc1077 TfLiteTensor* recovered_cell_weights; in Prepare() local
1080 &recovered_cell_weights)); in Prepare()
1081 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
1082 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
1084 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
1089 context->ResizeTensor(context, recovered_cell_weights, in Prepare()
Dlstm_eval.h163 TfLiteTensor* prod_scaling_factors, TfLiteTensor* recovered_cell_weights,
Dlstm.cc1535 TfLiteTensor* recovered_cell_weights; in Prepare() local
1538 &recovered_cell_weights)); in Prepare()
1539 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
1540 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
1542 if (!TfLiteIntArrayEqualsArray(recovered_cell_weights->dims, 1, in Prepare()
1547 context->ResizeTensor(context, recovered_cell_weights, in Prepare()