HomeSort by relevance Sort by last modified time
    Searched refs:ConvolveScratchSize (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 519 static int64 ConvolveScratchSize = GetCudnnWorkspaceLimit(
553 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
593 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
  /external/tensorflow/tensorflow/core/kernels/
conv_ops_3d.cc 369 static int64 ConvolveScratchSize = GetCudnnWorkspaceLimit(
405 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
441 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
conv_ops.cc 683 static int64 ConvolveScratchSize = GetCudnnWorkspaceLimit(
718 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
756 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
    [all...]

Completed in 101 milliseconds