Searched refs:dropout_state_allocator (Results 1 – 1 of 1) sorted by relevance
556 std::unique_ptr<CudnnRNNPersistentSpaceAllocator> dropout_state_allocator; member980 ScratchAllocator* dropout_state_allocator, in CreateRnnDescriptor() argument988 seed(), dropout_state_allocator); in CreateRnnDescriptor()1010 CudnnRNNPersistentSpaceAllocator* dropout_state_allocator = in GetCachedRnnDescriptor() local1012 rnn_state.dropout_state_allocator.reset(dropout_state_allocator); in GetCachedRnnDescriptor()1015 dropout_state_allocator, &rnn_state.rnn_desc); in GetCachedRnnDescriptor()1524 CudnnRnnAllocatorInTemp<uint8> dropout_state_allocator(context); in MaybeAutoTune() local1527 &dropout_state_allocator, &rnn_desc) in MaybeAutoTune()