Searched refs:CudnnRnnAllocatorInTemp (Results 1 – 1 of 1) sorted by relevance
360 class CudnnRnnAllocatorInTemp : public ScratchAllocator { class362 ~CudnnRnnAllocatorInTemp() override = default;364 explicit CudnnRnnAllocatorInTemp(OpKernelContext* context) in CudnnRnnAllocatorInTemp() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp1310 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeAndReturnAlgorithm()1524 CudnnRnnAllocatorInTemp<uint8> dropout_state_allocator(context); in MaybeAutoTune()1533 CudnnRnnAllocatorInTemp<T> reserve_space_allocator(context); in MaybeAutoTune()1534 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in MaybeAutoTune()1708 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeImpl()