Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector.cc106 lib_info_.reset(new FunctionLibraryApiInfo); in LoadFunctions()
107 TF_RETURN_IF_ERROR(lib_info_->Init(graph.library())); in LoadFunctions()
123 lib_info_->GetApiInfo(attr.second.func().name()) != nullptr) { in MaybeOptimizeFunctionCall()
129 lib_info_->GetApiInfo(node_def->op()) == nullptr) { in MaybeOptimizeFunctionCall()
146 TF_RETURN_IF_ERROR(lib_info_->GetEquivalentImplementations( 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()
160 TF_RETURN_IF_ERROR(lib_info_->GetEquivalentImplementations( in MaybeOptimizeFunctionCall()
163 const auto func_api_info = lib_info_->GetApiInfo(func_name); in MaybeOptimizeFunctionCall()
179 if (lib_info_->empty()) { in SelectImplementation()
Dimplementation_selector.h106 std::unique_ptr<FunctionLibraryApiInfo> lib_info_; variable