Searched defs:dropout_state_allocator (Results 1 – 1 of 1) sorted by relevance
563 std::unique_ptr<CudnnRNNSpaceAllocator> dropout_state_allocator; member1100 ScratchAllocator* dropout_state_allocator, in CreateRnnDescriptor()1132 CudnnRNNSpaceAllocator* dropout_state_allocator = in GetCachedRnnDescriptor() local1754 CudnnRnnAllocatorInTemp<uint8> dropout_state_allocator(context); in MaybeAutotune() local