Searched refs:ReachableDefinitions (Results 1 – 15 of 15) sorted by relevance
52 flib_def.ReachableDefinitions(*func_def); in ConstructFunctionGraph()169 *(gdef->mutable_library()) = flib_def.ReachableDefinitions(*gdef).ToProto(); in ConstructFunctionGraph()
116 .ReachableDefinitions(*output); in Optimize()
62 new Graph(graph->flib_def().ReachableDefinitions(graph_def))); in PartitionFunctionGraph()
724 *def.mutable_library() = lib_def->ReachableDefinitions(def).ToProto(); in InstantiateMultiDevice()752 lib_def->ReachableDefinitions(*fdef), std::move(ret_types)); in InstantiateMultiDevice()849 *def.mutable_library() = lib_def->ReachableDefinitions(def).ToProto(); in InstantiateMultiDevice()893 *def.mutable_library() = lib_def->ReachableDefinitions(def).ToProto(); in InstantiateMultiDevice()
95 func_lib_def.ReachableDefinitions(register_function->function_def()) in Instantiate()
507 FunctionLibraryDefinition ReachableDefinitions(const GraphDef& graph) const;508 FunctionLibraryDefinition ReachableDefinitions(const FunctionDef& func) const;
1356 TEST(FunctionLibraryDefinitionTest, ReachableDefinitions) { in TEST() argument1418 FunctionLibraryDefinition reachable_flib = flib.ReachableDefinitions(graph); in TEST()
1748 FunctionLibraryDefinition FunctionLibraryDefinition::ReachableDefinitions( in ReachableDefinitions() function in tensorflow::FunctionLibraryDefinition1753 FunctionLibraryDefinition FunctionLibraryDefinition::ReachableDefinitions( in ReachableDefinitions() function in tensorflow::FunctionLibraryDefinition
251 *function_body.mutable_library() = flib.ReachableDefinitions(func).ToProto(); in MakeGrapplerFunctionItem()
690 .ReachableDefinitions(graph); in OptimizeConsumeItem()
450 flib.ReachableDefinitions(optimized_graph); in PruneFunctionLibrary()
785 fld->ReachableDefinitions(graph_def); in PruneUnreachableFunctionsFromGraph()
1225 *gdef->mutable_library() = flib_def->ReachableDefinitions(*gdef).ToProto(); in Partition()
237 lib_def->ReachableDefinitions(*fdef)); in CreateFunctionLibraryDefinition()
4274 flib_def->ReachableDefinitions(graph_def); in RewriteTPUReplicateNode()