Searched refs:state_sizes_in_bytes (Results 1 – 1 of 1) sorted by relevance
1004 size_t state_sizes_in_bytes = 0; in Create() local1006 cudnnDropoutGetStatesSize(cudnn.handle(), &state_sizes_in_bytes)); in Create()1008 state_allocator->AllocateBytes(state_sizes_in_bytes)); in Create()