Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc354 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetCufftWorkspaceLimit() local
355 if (workspace_limit_in_mb_str != nullptr && in GetCufftWorkspaceLimit()
356 strcmp(workspace_limit_in_mb_str, "") != 0) { in GetCufftWorkspaceLimit()
362 << workspace_limit_in_mb_str; in GetCufftWorkspaceLimit()
Dconv_ops.cc554 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetDnnWorkspaceLimit() local
555 if (workspace_limit_in_mb_str != nullptr && in GetDnnWorkspaceLimit()
556 strcmp(workspace_limit_in_mb_str, "") != 0) { in GetDnnWorkspaceLimit()
558 if (strings::safe_strto64(workspace_limit_in_mb_str, in GetDnnWorkspaceLimit()
563 << workspace_limit_in_mb_str; in GetDnnWorkspaceLimit()