Searched refs:CudnnRnnAllocatorInOutput (Results 1 – 1 of 1) sorted by relevance
407 class CudnnRnnAllocatorInOutput : public ScratchAllocator { class409 ~CudnnRnnAllocatorInOutput() override {} in ~CudnnRnnAllocatorInOutput()410 CudnnRnnAllocatorInOutput(OpKernelContext* context, int output_index) in CudnnRnnAllocatorInOutput() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput1307 CudnnRnnAllocatorInOutput<T> reserve_space_allocator(context, 3); in ComputeAndReturnAlgorithm()