Searched refs:fw_scratch_buffer (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm.cc | 571 TfLiteTensor* fw_scratch_buffer; in Prepare() local
573 &fw_scratch_buffer)); in Prepare()
574 fw_scratch_buffer->type = input->type; in Prepare()
575 fw_scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
593 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_scratch_buffer, in Prepare()
1113 TfLiteTensor* fw_scratch_buffer; in Eval() local
1115 &fw_scratch_buffer)); in Eval()
1192 fw_scratch_buffer, fw_activation_state, fw_cell_state, fw_output); in Eval()
1289 fw_scratch_buffer, GetTemporary(context, node, kInputScalingFactors), in Eval()
|