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