Searched refs:fw_scratch_buffer (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A 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() 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/ |
D | BidirectionalSequenceLSTM.cpp | 482 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()
|