Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp483 std::vector<float> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval() local
517 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
607 std::vector<_Float16> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval() local
641 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()