Searched refs:state_memory (Results 1 – 1 of 1) sorted by relevance
943 DeviceMemory<uint8> state_memory; in Create() local948 SE_ASSIGN_OR_RETURN(state_memory, state_allocator->AllocateBytes( in Create()952 handle.get(), cudnn.handle(), dropout, state_memory.opaque(), in Create()953 state_memory.size(), seed)); in Create()