Home
last modified time | relevance | path

Searched refs:fw_scratch_buffer (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_lstm.cc571 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()
595 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_scratch_buffer, in Prepare()
1117 TfLiteTensor* fw_scratch_buffer; in Eval() local
1119 &fw_scratch_buffer)); in Eval()
1196 fw_scratch_buffer, fw_activation_state, fw_cell_state, fw_output, in Eval()
1294 fw_scratch_buffer, GetTemporary(context, node, kInputScalingFactors), in Eval()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp482 std::vector<float> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval() local
516 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
606 std::vector<_Float16> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval() local
640 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()