Searched defs:RnnScratchSpace (Results 1 – 1 of 1) sorted by relevance
554 struct RnnScratchSpace { struct555 std::unique_ptr<RnnDescriptor> rnn_desc;556 std::unique_ptr<CudnnRNNPersistentSpaceAllocator> dropout_state_allocator;