Lines Matching defs:lib_def
146 const FunctionLibraryDefinition* lib_def,
226 const FunctionLibraryDefinition* lib_def,
229 const FunctionLibraryDefinition* lib_def,
234 const FunctionLibraryDefinition* lib_def,
247 const FunctionLibraryDefinition* lib_def,
255 base_lib_def_(lib_def),
351 const NodeDef& ndef, const FunctionLibraryDefinition* lib_def,
368 if (lib_def->Find(ndef.op()) == nullptr) {
377 if (lib_def != base_lib_def_) {
378 options.overlay_lib = lib_def;
416 const FunctionLibraryDefinition* lib_def, FunctionBody** fbody) {
417 if (lib_def == base_lib_def_) {
418 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig_, fbody);
420 auto get_func_sig = [lib_def](const string& op, const OpDef** sig) {
421 return lib_def->LookUpOpDef(op, sig);
423 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig, fbody);
428 const NameAttrList& func, const FunctionLibraryDefinition* lib_def,
430 const FunctionDef* fdef = lib_def->Find(func.name());
444 FunctionDefToBody(grad_fdef, AttrSlice(&func.attr()), lib_def, g_body));
448 if (lib_def != base_lib_def_) {
449 options.overlay_lib = lib_def;
492 const FunctionLibraryDefinition* lib_def =
504 const string grad = lib_def->FindGradient(func.name());
508 TF_RETURN_IF_ERROR(InstantiateSymbolicGradient(func, lib_def, &fbody));
510 const FunctionDef* fdef = lib_def->Find(function_name);
514 TF_RETURN_IF_ERROR(FunctionDefToBody(*fdef, attrs, lib_def, &fbody));
599 const FunctionLibraryDefinition* lib_def;
603 lib_def = (*item)->overlay_lib;
605 if (!lib_def) {
606 lib_def = base_lib_def_;
608 std::unique_ptr<Graph> g(new Graph(lib_def));
621 if (lib_def == base_lib_def_) {
624 params.create_kernel = [this, lib_def](const NodeDef& ndef,
626 return CreateKernel(ndef, lib_def, kernel);
946 int graph_def_version, const FunctionLibraryDefinition* lib_def,
951 device_mgr, env, device, graph_def_version, lib_def, optimizer_options,
957 int graph_def_version, const FunctionLibraryDefinition* lib_def,
961 lib_def, optimizer_options,
1568 const FunctionLibraryDefinition* const lib_def,
1575 std::unique_ptr<Graph> graph(new Graph(lib_def));
