Home
last modified time | relevance | path

Searched defs:CudnnRnnAllocatorInOutput (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc407 class CudnnRnnAllocatorInOutput : public ScratchAllocator { class
410 CudnnRnnAllocatorInOutput(OpKernelContext* context, int output_index) in CudnnRnnAllocatorInOutput() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput