Searched refs:scratch_limit_in_mb (Results 1 – 2 of 2) sorted by relevance
357 int64 scratch_limit_in_mb = -1; in GetCufftWorkspaceLimit() local359 &scratch_limit_in_mb); in GetCufftWorkspaceLimit()364 return scratch_limit_in_mb * (1 << 20); in GetCufftWorkspaceLimit()
557 int64 scratch_limit_in_mb = -1; in GetDnnWorkspaceLimit() local559 &scratch_limit_in_mb)) { in GetDnnWorkspaceLimit()560 return scratch_limit_in_mb * (1 << 20); in GetDnnWorkspaceLimit()