Searched defs:dropout_state_allocator (Results 1 – 1 of 1) sorted by relevance
562 std::unique_ptr<CudnnRNNSpaceAllocator> dropout_state_allocator; member1099 ScratchAllocator* dropout_state_allocator, in CreateRnnDescriptor()1131 CudnnRNNSpaceAllocator* dropout_state_allocator = in GetCachedRnnDescriptor() local1752 CudnnRnnAllocatorInTemp<uint8> dropout_state_allocator(context); in MaybeAutotune() local