Searched defs:RnnScratchSpace (Results 1 – 1 of 1) sorted by relevance
561 struct RnnScratchSpace { struct562 std::unique_ptr<RnnDescriptor> rnn_desc;563 std::unique_ptr<CudnnRNNSpaceAllocator> dropout_state_allocator;