Searched refs:ConvolveScratchSize (Results 1 – 3 of 3) sorted by relevance
476 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in launch() local531 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()534 /*memory_limit=*/ConvolveScratchSize); in launch()564 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()624 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch()
351 inline int64 ConvolveScratchSize() {395 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);398 /*memory_limit=*/ConvolveScratchSize());700 DnnScratchAllocator scratch_allocator(ConvolveScratchSize(), context);
960 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in operator ()() local1019 /*memory_limit=*/ConvolveScratchSize); in operator ()()1020 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()1051 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()1115 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()()