Searched refs:CudnnRNNPersistentSpaceAllocator (Results 1 – 1 of 1) sorted by relevance
445 class CudnnRNNPersistentSpaceAllocator : public ScratchAllocator { class447 explicit CudnnRNNPersistentSpaceAllocator(OpKernelContext* context) in CudnnRNNPersistentSpaceAllocator() function in tensorflow::__anonebbbcdba0111::CudnnRNNPersistentSpaceAllocator450 ~CudnnRNNPersistentSpaceAllocator() override {} in ~CudnnRNNPersistentSpaceAllocator()556 std::unique_ptr<CudnnRNNPersistentSpaceAllocator> dropout_state_allocator;1010 CudnnRNNPersistentSpaceAllocator* dropout_state_allocator = in GetCachedRnnDescriptor()1011 new CudnnRNNPersistentSpaceAllocator(context); in GetCachedRnnDescriptor()