Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp535 std::vector<float> bw_scratch_buffer(getNumberOfElements(bw_scratch_shape_)); in Eval() local
571 bw_scratch_buffer.data(), params_.time_major, kBackwardSequence); in Eval()
660 std::vector<_Float16> bw_scratch_buffer(getNumberOfElements(bw_scratch_shape_)); in Eval() local
696 bw_scratch_buffer.data(), params_.time_major, kBackwardSequence); in Eval()