Home
last modified time | relevance | path

Searched refs:envvar_in_mb (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc352 int64 GetCufftWorkspaceLimit(const string& envvar_in_mb, in GetCufftWorkspaceLimit() argument
354 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetCufftWorkspaceLimit()
358 Status status = ReadInt64FromEnvVar(envvar_in_mb, default_value_in_bytes, in GetCufftWorkspaceLimit()
361 LOG(WARNING) << "Invalid value for env-var " << envvar_in_mb << ": " in GetCufftWorkspaceLimit()
Dconv_ops_gpu.h33 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb,
Dconv_ops.cc552 int64 GetDnnWorkspaceLimit(const string& envvar_in_mb, in GetDnnWorkspaceLimit() argument
554 const char* workspace_limit_in_mb_str = getenv(envvar_in_mb.c_str()); in GetDnnWorkspaceLimit()
562 LOG(WARNING) << "Invalid value for env-var " << envvar_in_mb << ": " in GetDnnWorkspaceLimit()