Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc912 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
914 EXPECT_EQ(lib_def.Find("XTimes16"), nullptr); in TEST()
924 auto found = lib_def.Find("XTimesTwo"); in TEST()
932 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
935 EXPECT_TRUE(!lib_def.LookUpOpDef("XTimes16", &op_def).ok()); in TEST()
937 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def)); in TEST()
943 TF_EXPECT_OK(lib_def.LookUp("XTimesTwo", &op_reg_data)); in TEST()
953 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
956 TF_EXPECT_OK(lib_def.AddFunctionDef(test::function::WXPlusB())); in TEST()
960 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &first)); in TEST()
[all …]
Dfunction.h289 explicit FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
291 const FunctionDefLibrary& lib_def);
332 Status AddLibrary(const FunctionDefLibrary& lib_def);
571 const string& function_name, const FunctionLibraryDefinition& lib_def,
Ddataset.h171 const FunctionLibraryDefinition* lib_def = in EnsureFunctionIsStateless() local
173 const FunctionDef* function_def = lib_def->Find(function_name); in EnsureFunctionIsStateless()
180 TF_RETURN_IF_ERROR(lib_def->LookUpOpDef(node_def.op(), &op_def)); in EnsureFunctionIsStateless()
Dfunction.cc1037 const FunctionDefLibrary& lib_def) { in AddLibrary() argument
1044 for (const FunctionDef& fdef : lib_def.function()) { in AddLibrary()
1054 for (const GradientDef& grad : lib_def.gradient()) { in AddLibrary()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc146 const FunctionLibraryDefinition* lib_def,
226 const FunctionLibraryDefinition* lib_def,
229 const FunctionLibraryDefinition* lib_def,
234 const FunctionLibraryDefinition* lib_def,
247 const FunctionLibraryDefinition* lib_def, in FunctionLibraryRuntimeImpl() argument
255 base_lib_def_(lib_def), in FunctionLibraryRuntimeImpl()
351 const NodeDef& ndef, const FunctionLibraryDefinition* lib_def, in CreateKernel() argument
368 if (lib_def->Find(ndef.op()) == nullptr) { in CreateKernel()
377 if (lib_def != base_lib_def_) { in CreateKernel()
378 options.overlay_lib = lib_def; in CreateKernel()
[all …]
Dprocess_function_library_runtime.cc30 const FunctionLibraryDefinition* lib_def, in ProcessFunctionLibraryRuntime() argument
34 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
40 lib_def, optimizer_options, this); in ProcessFunctionLibraryRuntime()
46 lib_def, optimizer_options, this); in ProcessFunctionLibraryRuntime()
52 const FunctionLibraryDefinition* lib_def, in ProcessFunctionLibraryRuntime() argument
57 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
62 nullptr, env, nullptr, graph_def_version, lib_def, optimizer_options, in ProcessFunctionLibraryRuntime()
68 device_mgr, env, d, graph_def_version, lib_def, optimizer_options, in ProcessFunctionLibraryRuntime()
Dfunction.h57 int graph_def_version, const FunctionLibraryDefinition* lib_def,
67 int graph_def_version, const FunctionLibraryDefinition* lib_def,
166 const FunctionLibraryDefinition* const lib_def,
Dprocess_function_library_runtime.h34 const FunctionLibraryDefinition* lib_def,
41 const FunctionLibraryDefinition* lib_def,
Dprocess_function_library_runtime_test.cc115 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST_F() local
121 lib_def.get(), opts, nullptr /* cluster_flr */)); in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc59 const FunctionLibraryDefinition& lib_def, in Instantiate() argument
63 return cluster_flr_->Instantiate(function_name, lib_def, attrs, options, in Instantiate()
68 const string& function_name, const FunctionLibraryDefinition& lib_def, in InstantiateAndRun() argument
73 TF_RETURN_IF_ERROR(cluster_flr_->Instantiate(function_name, lib_def, attrs, in InstantiateAndRun()
310 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
316 TF_EXPECT_OK(InstantiateAndRun("XTimesTwoInt32", lib_def, {}, in TEST_F()
325 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
331 TF_EXPECT_OK(InstantiateAndRun("Swap", lib_def, {{"T", DT_FLOAT}}, in TEST_F()
Dcluster_function_library_runtime.cc121 const string& function_name, const FunctionLibraryDefinition& lib_def, in Instantiate() argument
136 const FunctionDef* fdef = lib_def.Find(function_name); in Instantiate()
142 *gdef.mutable_library() = lib_def.ToProto(); in Instantiate()
Dgraph_mgr.cc124 item->lib_def.reset( in InitItem()
132 TF_RETURN_IF_ERROR(graph::ValidateGraphDef(gdef, *item->lib_def)); in InitItem()
137 item->lib_def.get(), graph_options.optimizer_options(), cluster_flr)); in InitItem()
184 optimization_options.flib_def = item->lib_def.get(); in InitItem()
Dcluster_function_library_runtime.h36 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
Dgraph_mgr.h130 std::unique_ptr<FunctionLibraryDefinition> lib_def; member
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc409 std::unique_ptr<FunctionLibraryDefinition> lib_def( in Encapsulate() local
413 std::unique_ptr<Graph> graph(new Graph(lib_def.get())); in Encapsulate()
422 &graph_out, lib_def.get()); in Encapsulate()
429 *library = lib_def->ToProto(); in Encapsulate()
490 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
492 GraphDefBuilder b2(GraphDefBuilder::kFailImmediately, lib_def.get()); in TEST()
496 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST()
550 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
552 GraphDefBuilder b2(GraphDefBuilder::kFailImmediately, lib_def.get()); in TEST()
557 NodeBuilder nb("F1", "F1", lib_def.get()); in TEST()
[all …]