Searched refs:local_dev_name (Results 1 – 1 of 1) sorted by relevance
392 const DeviceNameUtils::ParsedName& local_dev_name,744 DeviceNameUtils::ParsedName local_dev_name = device->parsed_name(); in SetAllocAttrs() local759 s = InferAllocAttr(n, e->dst(), local_dev_name, &attr); in SetAllocAttrs()783 const DeviceNameUtils::ParsedName& local_dev_name, in InferAllocAttr() argument798 if (!DeviceNameUtils::IsSameAddressSpace(parsed_src_name, local_dev_name)) { in InferAllocAttr()802 } else if ((local_dev_name.type == "CPU" || n->IsHostRecv()) && in InferAllocAttr()809 VLOG(2) << "default alloc case local type " << local_dev_name.type in InferAllocAttr()823 if (!DeviceNameUtils::IsSameAddressSpace(parsed_dst_name, local_dev_name)) { in InferAllocAttr()827 } else if ((local_dev_name.type == "CPU" || dst->IsHostSend()) && in InferAllocAttr()836 VLOG(2) << "default alloc case local type " << local_dev_name.type in InferAllocAttr()