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.cc403 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetCufftWorkspaceLimit() local
404 if (workspace_limit_in_mb_str != nullptr && in GetCufftWorkspaceLimit()
405 strcmp(workspace_limit_in_mb_str, "") != 0) { in GetCufftWorkspaceLimit()
411 << workspace_limit_in_mb_str; in GetCufftWorkspaceLimit()
Dconv_ops.cc622 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetDnnWorkspaceLimit() local
623 if (workspace_limit_in_mb_str != nullptr && in GetDnnWorkspaceLimit()
624 strcmp(workspace_limit_in_mb_str, "") != 0) { in GetDnnWorkspaceLimit()
626 if (strings::safe_strto64(workspace_limit_in_mb_str, in GetDnnWorkspaceLimit()
631 << workspace_limit_in_mb_str; in GetDnnWorkspaceLimit()