Searched refs:cell_state_quantized (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 371 TfLiteTensor* cell_state_quantized = in Prepare() local
373 cell_state_quantized->type = input_to_output_weights->type; in Prepare()
374 cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
375 if (!TfLiteIntArrayEqual(cell_state_quantized->dims, cell_state->dims)) { in Prepare()
379 context->ResizeTensor(context, cell_state_quantized, in Prepare()
525 TfLiteTensor* cell_state_quantized = in Eval() local
552 cell_state_quantized, activation_state, cell_state, output); in Eval()
|
| D | lstm.cc | 443 TfLiteTensor* cell_state_quantized = in Prepare() local
445 cell_state_quantized->type = input_to_output_weights->type; in Prepare()
446 cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
447 if (!TfLiteIntArrayEqual(cell_state_quantized->dims, cell_state->dims)) { in Prepare()
451 context->ResizeTensor(context, cell_state_quantized, in Prepare()
604 TfLiteTensor* cell_state_quantized = in Eval() local
630 cell_state_quantized, activation_state, cell_state, output); in Eval()
|
| D | lstm_eval.h | 84 TfLiteTensor* cell_state_quantized, TfLiteTensor* output_state,
|
| D | lstm_eval.cc | 1097 TfLiteTensor* cell_state_quantized, TfLiteTensor* output_state, in EvalHybrid()
1232 GetInt8DataPtr(cell_state_quantized, is_uint8_hybrid); in EvalHybrid()
|