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.cc233 lib_info_ = absl::make_unique<FunctionLibraryApiInfo>(); in LoadFunctions()
234 TF_RETURN_IF_ERROR(lib_info_->Init(graph.library())); in LoadFunctions()
252 lib_info_->GetApiInfo(attr.second.func().name()) != nullptr) { in MaybeOptimizeFunctionCall()
258 lib_info_->GetApiInfo(node_def->op()) == nullptr) { in MaybeOptimizeFunctionCall()
276 TF_RETURN_IF_ERROR(lib_info_->GetEquivalentImplementations( in MaybeOptimizeFunctionCall()
279 const auto& func_api_info = lib_info_->GetApiInfo(func_name); in MaybeOptimizeFunctionCall()
288 if (lib_info_->GetApiInfo(node_def->op()) != nullptr && in MaybeOptimizeFunctionCall()
291 TF_RETURN_IF_ERROR(lib_info_->GetEquivalentImplementations( in MaybeOptimizeFunctionCall()
294 const auto func_api_info = lib_info_->GetApiInfo(func_name); in MaybeOptimizeFunctionCall()
379 if (lib_info_->empty()) { in SelectImplementation()
Dimplementation_selector.h194 std::unique_ptr<FunctionLibraryApiInfo> lib_info_; variable