Home
last modified time | relevance | path

Searched refs:ConvolveBackwardDataScratchSize (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc356 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in operator ()() local
359 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()()
410 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in operator ()()
414 /*memory_limit=*/ConvolveBackwardDataScratchSize); in operator ()()
442 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()()
Dconv_grad_ops_3d.cc1394 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in Compute() local
1444 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1448 /*memory_limit=*/ConvolveBackwardDataScratchSize); in Compute()
1480 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1521 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()