Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector.cc123 lib_info_->GetApiInfo(attr.second.func().name()) != nullptr) { in MaybeOptimizeFunctionCall()
129 lib_info_->GetApiInfo(node_def->op()) == nullptr) { in MaybeOptimizeFunctionCall()
149 const auto& func_api_info = lib_info_->GetApiInfo(func_name); in MaybeOptimizeFunctionCall()
158 if (lib_info_->GetApiInfo(node_def->op()) != nullptr) { in MaybeOptimizeFunctionCall()
163 const auto func_api_info = lib_info_->GetApiInfo(func_name); in MaybeOptimizeFunctionCall()
Dfunction_api_info_test.cc121 auto* info = lib_api_info.GetApiInfo(func_name); in GetInterfaceName()
128 auto* info = lib_api_info.GetApiInfo(func_name); in GetPreferredDevice()
Dfunction_api_info.h82 const FunctionApiInfo* GetApiInfo(const string& function_name) const;
Dfunction_api_info.cc229 const FunctionApiInfo* FunctionLibraryApiInfo::GetApiInfo( in GetApiInfo() function in tensorflow::grappler::FunctionLibraryApiInfo