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.cc926 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in operator ()() local
929 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()()
962 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in operator ()()
Dconv_grad_ops_3d.cc1335 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in Compute() local
1370 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()
1407 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute()