Searched refs:envvar_in_mb (Results 1 – 3 of 3) sorted by relevance
401 int64 GetCufftWorkspaceLimit(const string& envvar_in_mb, in GetCufftWorkspaceLimit() argument403 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetCufftWorkspaceLimit()407 Status status = ReadInt64FromEnvVar(envvar_in_mb, default_value_in_bytes, in GetCufftWorkspaceLimit()410 LOG(WARNING) << "Invalid value for env-var " << envvar_in_mb << ": " in GetCufftWorkspaceLimit()
45 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb,
620 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb, in GetDnnWorkspaceLimit() argument622 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetDnnWorkspaceLimit()630 LOG(WARNING) << "Invalid value for env-var " << envvar_in_mb << ": " in GetDnnWorkspaceLimit()