Searched defs:CudnnRnnAllocatorInTemp (Results 1 – 1 of 1) sorted by relevance
360 class CudnnRnnAllocatorInTemp : public ScratchAllocator { class364 explicit CudnnRnnAllocatorInTemp(OpKernelContext* context) in CudnnRnnAllocatorInTemp() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp