Searched refs:prod_scaling_factors (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm.cc | 791 TfLiteTensor* prod_scaling_factors; in Prepare() local
794 &prod_scaling_factors)); in Prepare()
795 prod_scaling_factors->type = kTfLiteFloat32; in Prepare()
796 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
797 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
802 context->ResizeTensor(context, prod_scaling_factors, in Prepare()
1239 TfLiteTensor* prod_scaling_factors; in Eval() local
1242 &prod_scaling_factors)); in Eval()
1292 prod_scaling_factors, recovered_cell_weights, input_quantized, in Eval()
1333 prod_scaling_factors, recovered_cell_weights, input_quantized, in Eval()
|
| D | unidirectional_sequence_lstm.cc | 1058 TfLiteTensor* prod_scaling_factors; in Prepare() local
1061 &prod_scaling_factors)); in Prepare()
1062 prod_scaling_factors->type = kTfLiteFloat32; in Prepare()
1063 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
1064 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
1069 context->ResizeTensor(context, prod_scaling_factors, in Prepare()
|
| D | lstm_eval.h | 163 TfLiteTensor* prod_scaling_factors, TfLiteTensor* recovered_cell_weights,
|
| D | lstm.cc | 1516 TfLiteTensor* prod_scaling_factors; in Prepare() local
1519 &prod_scaling_factors)); in Prepare()
1520 prod_scaling_factors->type = kTfLiteFloat32; in Prepare()
1521 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
1522 if (!TfLiteIntArrayEqualsArray(prod_scaling_factors->dims, 1, in Prepare()
1527 context->ResizeTensor(context, prod_scaling_factors, in Prepare()
|
| D | lstm_eval.cc | 1905 TfLiteTensor* prod_scaling_factors, TfLiteTensor* recovered_cell_weights, in EvalHybrid()
2034 GetTensorData<float>(prod_scaling_factors), in EvalHybrid()
2133 GetTensorData<float>(prod_scaling_factors), in EvalHybrid()
|