Home
last modified time | relevance | path

Searched refs:lib_def (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc973 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
974 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
976 EXPECT_FALSE(lib_def.Contains("XTimes16")); in TEST()
977 EXPECT_TRUE(lib_def.Contains("XTimesTwo")); in TEST()
981 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
982 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
984 EXPECT_EQ(lib_def.Find("XTimes16"), nullptr); in TEST()
986 auto found = lib_def.Find("XTimesTwo"); in TEST()
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
993 TF_CHECK_OK(lib_def.AddFunctionDef(test::function::XTimesTwo())); in TEST()
[all …]
Ddataset.cc274 const FunctionLibraryDefinition& lib_def) { in AddFunction() argument
280 const FunctionDef* f_def = lib_def.Find(function_name); in AddFunction()
287 const string gradient_func = lib_def.FindGradient(function_name); in AddFunction()
298 TF_RETURN_IF_ERROR(lib_def.LookUp(node_def.op(), &op_reg_data)); in AddFunction()
300 TF_RETURN_IF_ERROR(AddFunction(ctx, op_reg_data->op_def.name(), lib_def)); in AddFunction()
304 TF_RETURN_IF_ERROR(AddAttrFunctions(ctx, pair.second, lib_def)); in AddFunction()
310 TF_RETURN_IF_ERROR(AddAttrFunctions(ctx, iter->second, lib_def)); in AddFunction()
Dfunction.h383 FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
385 const FunctionDefLibrary& lib_def);
452 Status AddLibrary(const FunctionDefLibrary& lib_def) TF_LOCKS_EXCLUDED(mu_);
691 const FunctionLibraryDefinition* lib_def = nullptr; member
973 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
Dgraph_to_functiondef.h65 const FunctionLibraryDefinition* lib_def, std::unique_ptr<Graph>* graph,
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_def_utils.cc31 const FunctionLibraryDefinition* const lib_def, in FunctionDefToBodyHelper() argument
38 auto graph = absl::make_unique<Graph>(lib_def); in FunctionDefToBodyHelper()
56 lib_def->GetStackTraces(fdef.signature().name()); in FunctionDefToBodyHelper()
77 const FunctionLibraryDefinition* lib_def, in FunctionDefToBodyHelper() argument
79 const auto get_func_sig = [&lib_def](const string& op, const OpDef** sig) { in FunctionDefToBodyHelper()
80 return lib_def->LookUpOpDef(op, sig); in FunctionDefToBodyHelper()
82 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig, fbody); in FunctionDefToBodyHelper()
Dfunction.cc149 const FunctionLibraryDefinition* lib_def) in FunctionLibraryRuntimeOverlay() argument
150 : base_flr_(base_flr), lib_def_(lib_def) {} in FunctionLibraryRuntimeOverlay()
210 if (!options.lib_def && lib_def_) { in Instantiate()
212 options_copy.lib_def = lib_def_; in Instantiate()
326 const FunctionLibraryDefinition* lib_def,
410 const FunctionLibraryDefinition* lib_def = nullptr; // Not owned. member
434 const FunctionLibraryDefinition* lib_def,
439 const FunctionLibraryDefinition* lib_def,
460 int graph_def_version, const FunctionLibraryDefinition* lib_def, in FunctionLibraryRuntimeImpl() argument
470 base_lib_def_(lib_def), in FunctionLibraryRuntimeImpl()
[all …]
Dfunction_def_utils.h35 const FunctionLibraryDefinition* lib_def,
43 const FunctionLibraryDefinition* lib_def,
Dprocess_function_library_runtime.cc65 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, in DistributedInit() argument
78 parent->Instantiate(function_name, lib_def, attrs, options, &local_handle_, in DistributedInit()
87 int graph_def_version, const FunctionLibraryDefinition* lib_def, in ProcessFunctionLibraryRuntime() argument
97 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
634 const FunctionLibraryDefinition* lib_def, std::unique_ptr<Graph>* graph, in GetGraphAndArgRets() argument
640 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*fdef, attrs, lib_def, &fbody)); in GetGraphAndArgRets()
700 const FunctionLibraryDefinition* lib_def = in InstantiateMultiDevice() local
701 options.lib_def == nullptr ? lib_def_ : options.lib_def; in InstantiateMultiDevice()
703 const FunctionDef* fdef = lib_def->Find(function_name); in InstantiateMultiDevice()
706 "\" in function library: ", lib_def); in InstantiateMultiDevice()
[all …]
Dprocess_function_library_runtime.h68 int graph_def_version, const FunctionLibraryDefinition* lib_def,
266 FunctionLibraryDefinition&& lib_def, in MultiDeviceFunctionData()
271 lib_def_(std::move(lib_def)), in MultiDeviceFunctionData()
424 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
Dfunction.h62 const FunctionLibraryDefinition* lib_def, thread::ThreadPool* thread_pool,
Dfunction_utils.h102 bool IsFunctionCall(const FunctionLibraryDefinition& lib_def, const Node& n);
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc57 const FunctionLibraryDefinition& lib_def, GraphDef* g, in ConstructFunctionGraphHelper() argument
60 sig, attrs, options, lib_def, g, send_keys, recv_keys); in ConstructFunctionGraphHelper()
64 const FunctionLibraryDefinition& lib_def, in Instantiate() argument
69 cluster_flr_->Instantiate(function_name, lib_def, attrs, options, in Instantiate()
74 const string& function_name, const FunctionLibraryDefinition& lib_def, in InstantiateAndRun() argument
81 cluster_flr_->Instantiate(function_name, lib_def, attrs, options, &handle, in InstantiateAndRun()
123 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
129 lib_def, &actual, &send_keys, &recv_keys)); in TEST_F()
382 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
388 TF_EXPECT_OK(InstantiateAndRun("XTimesTwoInt32", lib_def, {}, in TEST_F()
[all …]
Dcluster_function_library_runtime.cc182 const string& function_name, const FunctionLibraryDefinition& lib_def, in Instantiate() argument
206 auto construct_graph_fn = [&](const FunctionLibraryDefinition* lib_def) { in Instantiate() argument
207 const FunctionDef* fdef = lib_def->Find(function_name); in Instantiate()
209 TF_RETURN_IF_ERROR(ConstructFunctionGraph(sig, attrs, options, *lib_def, in Instantiate()
214 if (options.lib_def) { in Instantiate()
215 s = construct_graph_fn(options.lib_def); in Instantiate()
217 s = construct_graph_fn(&lib_def); in Instantiate()
/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_utils.cc165 std::unique_ptr<FunctionLibraryDefinition> lib_def = nullptr; in RewriteDataset() local
167 ctx->function_library()->Clone(&lib_def, &pflr, &flr, true)); in RewriteDataset()
171 TF_RETURN_IF_ERROR(AddToFunctionLibrary(lib_def.get(), graph_def.library())); in RewriteDataset()
185 lib_def = lib_def.release(), in RewriteDataset()
188 std::unique_ptr<FunctionLibraryDefinition> lib_def_owner(lib_def); in RewriteDataset()
201 Status s = HashNode(graph_def, *node_def, *lib_def, &hash); in RewriteDataset()
Dserialization_utils.cc30 FunctionLibraryDefinition lib_def(OpRegistry::Global(), graph_def.library()); in FindStatefulOps() local
36 if (!IsNodeStateful(lib_def, node).ok()) { in FindStatefulOps()
45 if (!IsNodeStateful(lib_def, node).ok()) { in FindStatefulOps()
Dcaptured_function.cc228 const FunctionLibraryDefinition* lib_def, const string& func_name, in CreateFunctionLibraryDefinition() argument
230 DCHECK(lib_def != nullptr); in CreateFunctionLibraryDefinition()
231 const FunctionDef* fdef = lib_def->Find(func_name); in CreateFunctionLibraryDefinition()
237 lib_def->ReachableDefinitions(*fdef)); in CreateFunctionLibraryDefinition()
238 return (*result)->CopyFunctionDefFrom(func_name, *lib_def); in CreateFunctionLibraryDefinition()
265 Status LookupFunction(const FunctionLibraryDefinition& lib_def, in LookupFunction() argument
267 *fdef = lib_def.Find(name); in LookupFunction()
271 " in function library: ", lib_def.ToProto().DebugString()); in LookupFunction()
510 TF_RETURN_IF_ERROR(LookupFunction(*(*out_metadata)->lib_def(), in Create()
584 b->AddFunction(ctx, metadata_->func().name(), *metadata_->lib_def())); in AddToGraph()
[all …]
Dcaptured_function.h95 const FunctionLibraryDefinition* lib_def() const { return lib_def_.get(); } in lib_def() function
172 const FunctionLibraryDefinition* lib_def() const { in lib_def() function
173 return metadata_->lib_def(); in lib_def()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc484 std::unique_ptr<FunctionLibraryDefinition> lib_def( in Encapsulate() local
488 std::unique_ptr<Graph> graph(new Graph(lib_def.get())); in Encapsulate()
504 TF_GRAPH_DEF_VERSION, lib_def.get(), opts, in Encapsulate()
512 &graph_out, lib_def.get()); in Encapsulate()
534 graph_out.get(), flr, lib_def.get(), &modified); in Encapsulate()
541 *library = lib_def->ToProto(); in Encapsulate()
616 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
618 GraphDefBuilder b2(GraphDefBuilder::kFailImmediately, lib_def.get()); in TEST()
622 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST()
676 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc29 const FunctionLibraryDefinition& lib_def, in RunShapeInference() argument
36 TF_RETURN_IF_ERROR(lib_def.LookUp(ndef.op(), &op_reg_data)); in RunShapeInference()
Dshape_inference.h29 const FunctionLibraryDefinition& lib_def,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils_test.cc804 FunctionLibraryDefinition lib_def(OpRegistry::Global(), in TEST() local
807 EXPECT_FALSE(IsFunctionStateful(lib_def, *function)); in TEST()
810 EXPECT_TRUE(IsNodeStateful(lib_def, *nodeA)); in TEST()
835 EXPECT_FALSE(IsNodeStateful(lib_def, node)); in TEST()
838 EXPECT_TRUE(IsNodeStateful(lib_def, node)); in TEST()
839 EXPECT_FALSE(IsNodeStateful(lib_def, node, true)); in TEST()
Dslack_test.cc114 FunctionDefLibrary lib_def; in TEST() local
115 FunctionDef *fdef = lib_def.add_function(); in TEST()
155 FunctionLibraryDefinition flib(OpRegistry::Global(), lib_def); in TEST()
Dvectorization_utils_test.cc262 FunctionLibraryDefinition lib_def(OpRegistry::Global(), lib); in TEST() local
264 lib_def.Find(map_defun_node.attr().at("f").func().name()); in TEST()
479 FunctionLibraryDefinition lib_def(OpRegistry::Global(), lib); in TEST() local
481 lib_def.Find(map_defun_node.attr().at("f").func().name()); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.cc49 const string& function_name, const FunctionLibraryDefinition& lib_def, in Instantiate() argument
82 options.lib_def ? *options.lib_def : lib_def; in Instantiate()
Dremote_execute_node.h44 const NodeDef& ndef, FunctionLibraryDefinition* lib_def, in RemoteExecuteNode() argument
55 lib_def_(lib_def), in RemoteExecuteNode()

12