Home
last modified time | relevance | path

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

/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()