Searched refs:platform_str (Results 1 – 2 of 2) sorted by relevance
46 string platform_str = absl::AsciiStrToLower(name); in CanonicalPlatformName() local48 if (platform_str == "cpu") { in CanonicalPlatformName()49 platform_str = "host"; in CanonicalPlatformName()52 if (platform_str == "gpu") { in CanonicalPlatformName()53 platform_str = "cuda"; in CanonicalPlatformName()55 return platform_str; in CanonicalPlatformName()143 string platform_str = CanonicalPlatformName(platform_name); in GetPlatform() local146 if (absl::AsciiStrToLower(platform->Name()) == platform_str) { in GetPlatform()158 string platform_str = CanonicalPlatformName(platform_name); in GetPlatformExceptFor() local
34 string platform_str; in RealMain() local36 tensorflow::Flag("platform", &platform_str, in RealMain()52 if (!platform_str.empty()) { in RealMain()53 platform = PlatformUtil::GetPlatform(platform_str).ValueOrDie(); in RealMain()