Searched refs:state_memory (Results 1 – 1 of 1) sorted by relevance
1176 DeviceMemory<uint8> state_memory; in Create() local1181 TF_ASSIGN_OR_RETURN(state_memory, in Create()1185 handle.get(), cudnn.handle(), dropout, state_memory.opaque(), in Create()1186 state_memory.size(), seed)); in Create()