Searched defs:dropout_state_allocator (Results 1 – 1 of 1) sorted by relevance
560 std::unique_ptr<CudnnRNNPersistentSpaceAllocator> dropout_state_allocator; member1058 ScratchAllocator* dropout_state_allocator, in CreateRnnDescriptor()1090 CudnnRNNPersistentSpaceAllocator* dropout_state_allocator = in GetCachedRnnDescriptor() local1710 CudnnRnnAllocatorInTemp<uint8> dropout_state_allocator(context); in MaybeAutoTune() local