Searched refs:loadGRUCell (Results 1 – 1 of 1) sorted by relevance
3885 auto loadGRUCell = [&](bool forward, std::vector<NodeValue> &Yslices, in createOnnxGRU() local4035 loadGRUCell(/* forward */ true, forwardYslices, forwardHslice); in createOnnxGRU()4043 loadGRUCell(/* forward */ false, backwardYslices, backwardHslice); in createOnnxGRU()