Searched defs:CudnnRnnAllocatorInOutput (Results 1 – 1 of 1) sorted by relevance
407 class CudnnRnnAllocatorInOutput : public ScratchAllocator { class410 CudnnRnnAllocatorInOutput(OpKernelContext* context, int output_index) in CudnnRnnAllocatorInOutput() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput