Searched refs:ConvolveScratchSize (Results 1 – 4 of 4) sorted by relevance
412 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in launch() local454 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()484 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
490 inline int64 ConvolveScratchSize() {526 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);806 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);
821 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in operator ()() local864 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()894 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()
657 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in launch() local718 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()753 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()