Searched defs: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::CudnnRNNPersistentSpaceAllocator