Searched refs:state_memory (Results 1 – 1 of 1) sorted by relevance
1002 DeviceMemory<uint8> state_memory; in Create() local1007 SE_ASSIGN_OR_RETURN(state_memory, in Create()1011 handle.get(), cudnn.handle(), dropout, state_memory.opaque(), in Create()1012 state_memory.size(), seed)); in Create()