Home
last modified time | relevance | path

Searched refs:func_api_info (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector.cc279 const auto& func_api_info = lib_info_->GetApiInfo(func_name); in MaybeOptimizeFunctionCall() local
280 if (func_api_info->preferred_device() == parsed_name.type) { in MaybeOptimizeFunctionCall()
282 TF_RETURN_IF_ERROR(UpdateNodeDef(node_view, func_name, *func_api_info)); in MaybeOptimizeFunctionCall()
294 const auto func_api_info = lib_info_->GetApiInfo(func_name); in MaybeOptimizeFunctionCall() local
295 if (func_api_info->preferred_device() == parsed_name.type) { in MaybeOptimizeFunctionCall()