Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h33 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb,
Dconv_ops.cc552 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb, in GetDnnWorkspaceLimit() function
821 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in operator ()()
Dconv_ops_3d.cc412 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in launch()
Dconv_grad_ops_3d.cc1335 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in Compute()
1741 static int64 ConvolveBackwardFilterScratchSize = GetDnnWorkspaceLimit( in Compute()
Dconv_ops_fused_impl.h491 static int64 convolve_scratch_size = GetDnnWorkspaceLimit(
Dconv_grad_filter_ops.cc815 static int64 ConvolveBackwardFilterScratchSize = GetDnnWorkspaceLimit( in operator ()()
Dconv_grad_input_ops.cc926 static int64 ConvolveBackwardDataScratchSize = GetDnnWorkspaceLimit( in operator ()()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc657 static int64 ConvolveScratchSize = GetDnnWorkspaceLimit( in launch()