Searched refs:preferred_device (Results 1 – 5 of 5) sorted by relevance
48 const string& preferred_device, in PopulateFunction() argument62 if (!preferred_device.empty()) in PopulateFunction()63 (*func_attr)["api_preferred_device"].set_s(preferred_device); in PopulateFunction()130 return info->preferred_device(); in GetPreferredDevice()
150 if (func_api_info->preferred_device() == parsed_name.type) { in MaybeOptimizeFunctionCall()164 if (func_api_info->preferred_device() == parsed_name.type) { in MaybeOptimizeFunctionCall()
45 const string& preferred_device() const;
65 const string& FunctionApiInfo::preferred_device() const { in preferred_device() function in tensorflow::grappler::FunctionApiInfo
816 def _generate_defun_backend(unique_api_name, preferred_device, func): argument819 _DEFUN_DEVICE_ATTRIBUTE: preferred_device,