Searched defs:workspace_allocator (Results 1 – 3 of 3) sorted by relevance
1503 ScratchAllocator* workspace_allocator, in CreateRnnWorkspace()1548 ScratchAllocator* workspace_allocator) { in DoRnnForwardImpl()1672 ScratchAllocator* workspace_allocator) { in DoRnnBackwardImpl()1844 ScratchAllocator* workspace_allocator) { in DoRnnForward()1886 ScratchAllocator* workspace_allocator) { in DoRnnForward()1929 ScratchAllocator* workspace_allocator) { in DoRnnForward()1979 ScratchAllocator* workspace_allocator) { in DoRnnBackward()2031 ScratchAllocator* workspace_allocator) { in DoRnnBackward()2084 ScratchAllocator* workspace_allocator) { in DoRnnBackward()
885 CudnnRNNWorkspaceAllocator workspace_allocator(context); in Compute() local1089 CudnnRNNWorkspaceAllocator workspace_allocator(context); in Compute() local
4698 ScratchAllocator *workspace_allocator) { in ThenRnnForward()4730 ScratchAllocator *workspace_allocator) { in ThenRnnForward()4763 ScratchAllocator *workspace_allocator) { in ThenRnnForward()4803 ScratchAllocator *workspace_allocator) { in ThenRnnBackward()4844 ScratchAllocator *workspace_allocator) { in ThenRnnBackward()4886 ScratchAllocator *workspace_allocator) { in ThenRnnBackward()