Searched defs:device_string (Results 1 – 13 of 13) sorted by relevance
13 DeviceType parse_type(const std::string& device_string) { in parse_type()68 Device::Device(const std::string& device_string) : Device(Type::CPU) { in Device()
131 std::string device_string = GetDeviceString(params.nccl_params); in RunNcclCollective() local144 absl::string_view device_string, int64_t current_id) { in RunCollectivePermute()
2037 std::string device_string = NcclCollectiveThunk::GetDeviceString(params); in operator ()() local
10 std::string device_string; member
54 string device_string = "CPU"; in PersistentMemoryTrackingTest() local
91 bool IsPropagatableDevice(StringPiece device_string) { in IsPropagatableDevice()
64 std::string device_string; in GetDeviceNumber() local
83 std::string device_string = serialized_arg_val["type"].get<std::string>(); in prefill_stack_with_static_arguments() local
53 def set_cpu0(device_string): argument
95 bool AudioPowerUsage::deviceFromString(const std::string& device_string, int32_t& device) { in deviceFromString()
1066 def check_device(path, device_string): argument
809 auto device_string = stack_.back().toTupleRef().elements().at(0); in readGlobal() local
1368 bool IsTpuDevice(StringPiece device_string) { in IsTpuDevice()