Home
last modified time | relevance | path

Searched refs:GetDnnWorkspaceLimit (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc102 static int64 PoolingScratchSize = GetDnnWorkspaceLimit( in Compute()
244 static int64 PoolingScratchSize = GetDnnWorkspaceLimit( in Compute()
Dconv_ops_gpu.h45 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb,
Dpooling_ops_common.cc390 static int64 PoolingScratchSize = GetDnnWorkspaceLimit( in Compute()
708 static int64 PoolingScratchSize = GetDnnWorkspaceLimit( in Compute()
Dconv_ops.cc620 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb, in GetDnnWorkspaceLimit() function
960 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in operator ()()
Dconv_grad_input_ops.cc356 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in operator ()()
Dconv_ops_3d.cc476 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in launch()
Dlrn_op.cc599 static int64 NormalizeBackwardScratchSize = GetDnnWorkspaceLimit( in launch()
Dconv_ops_fused_impl.h352 static int64 convolve_scratch_size = GetDnnWorkspaceLimit(
Dconv_grad_ops_3d.cc1394 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in Compute()
1893 static int64 ConvolveBackwardFilterScratchSize = GetDnnWorkspaceLimit( in Compute()
Dconv_grad_filter_ops.cc952 static int64 ConvolveBackwardFilterScratchSize = GetDnnWorkspaceLimit( in operator ()()