Searched defs: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()1010 CudnnRNNPersistentSpaceAllocator* dropout_state_allocator = in GetCachedRnnDescriptor() local1524 CudnnRnnAllocatorInTemp<uint8> dropout_state_allocator(context); in MaybeAutoTune() local