Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc972 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
974 EXPECT_EQ(lib_def.Find("XTimes16"), nullptr); in TEST()
984 auto found = lib_def.Find("XTimesTwo"); in TEST()
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
995 EXPECT_TRUE(!lib_def.LookUpOpDef("XTimes16", &op_def).ok()); in TEST()
997 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def)); in TEST()
1003 TF_EXPECT_OK(lib_def.LookUp("XTimesTwo", &op_reg_data)); in TEST()
1013 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
1016 TF_EXPECT_OK(lib_def.AddFunctionDef(test::function::WXPlusB())); in TEST()
1020 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &first)); in TEST()
[all …]
Dfunction.h315 FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
317 const FunctionDefLibrary& lib_def);
374 Status AddLibrary(const FunctionDefLibrary& lib_def) LOCKS_EXCLUDED(mu_);
729 const string& function_name, const FunctionLibraryDefinition& lib_def,
Dfunction.cc1170 const FunctionDefLibrary& lib_def) { in AddLibrary() argument
1178 for (const FunctionDef& fdef : lib_def.function()) { in AddLibrary()
1188 for (const GradientDef& grad : lib_def.gradient()) { in AddLibrary()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc60 const FunctionLibraryDefinition& lib_def, in Instantiate() argument
64 return cluster_flr_->Instantiate(function_name, lib_def, attrs, options, in Instantiate()
69 const string& function_name, const FunctionLibraryDefinition& lib_def, in InstantiateAndRun() argument
74 TF_RETURN_IF_ERROR(cluster_flr_->Instantiate(function_name, lib_def, attrs, in InstantiateAndRun()
311 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
317 TF_EXPECT_OK(InstantiateAndRun("XTimesTwoInt32", lib_def, {}, in TEST_F()
326 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
332 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
138 const FunctionDef* fdef = lib_def.Find(function_name); in Instantiate()
144 *gdef.mutable_library() = lib_def.ToProto(); in Instantiate()
Dgraph_mgr.cc130 item->lib_def.reset( in InitItem()
138 TF_RETURN_IF_ERROR(graph::ValidateGraphDef(gdef, *item->lib_def)); in InitItem()
143 item->lib_def.get(), graph_options.optimizer_options(), in InitItem()
191 optimization_options.flib_def = item->lib_def.get(); in InitItem()
Dcluster_function_library_runtime.h38 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
Dgraph_mgr.h131 std::unique_ptr<FunctionLibraryDefinition> lib_def; member
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc309 const FunctionLibraryDefinition* lib_def,
401 const FunctionLibraryDefinition* lib_def,
404 const FunctionLibraryDefinition* lib_def,
409 const FunctionLibraryDefinition* lib_def,
426 const FunctionLibraryDefinition* lib_def, in FunctionLibraryRuntimeImpl() argument
435 base_lib_def_(lib_def), in FunctionLibraryRuntimeImpl()
538 const NodeDef& ndef, const FunctionLibraryDefinition* lib_def, in CreateKernel() argument
555 if (lib_def->Find(ndef.op()) == nullptr) { in CreateKernel()
564 if (lib_def != base_lib_def_) { in CreateKernel()
565 options.overlay_lib = lib_def; in CreateKernel()
[all …]
Dfunction.h58 int graph_def_version, const FunctionLibraryDefinition* lib_def,
68 int graph_def_version, const FunctionLibraryDefinition* lib_def,
282 bool IsFunctionCall(const FunctionLibraryDefinition& lib_def, const Node& n);
288 const FunctionLibraryDefinition* const lib_def,
Dprocess_function_library_runtime.cc47 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, in DistributedInit() argument
52 init_result_ = parent->Instantiate(function_name, lib_def, attrs, options, in DistributedInit()
60 const FunctionLibraryDefinition* lib_def, in ProcessFunctionLibraryRuntime() argument
66 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
85 const FunctionLibraryDefinition* lib_def, in ProcessFunctionLibraryRuntime() argument
92 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
480 const FunctionLibraryDefinition* lib_def, in GetGraphAndRets() argument
484 auto get_func_sig = [lib_def](const string& op, const OpDef** sig) { in GetGraphAndRets()
485 return lib_def->LookUpOpDef(op, sig); in GetGraphAndRets()
490 FunctionDefToBodyHelper(*fdef, attrs, lib_def, get_func_sig, &tmp_fbody)); in GetGraphAndRets()
[all …]
Dprocess_function_library_runtime.h36 const FunctionLibraryDefinition* lib_def,
44 const FunctionLibraryDefinition* lib_def,
274 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
Dprocess_function_library_runtime_test.cc47 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, in Instantiate() argument
222 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST_F() local
228 lib_def.get(), opts, nullptr, nullptr /* cluster_flr */)); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc493 std::unique_ptr<FunctionLibraryDefinition> lib_def( in Encapsulate() local
497 std::unique_ptr<Graph> graph(new Graph(lib_def.get())); in Encapsulate()
512 device_mgr.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def.get(), in Encapsulate()
520 /*reuse_existing_functions=*/false, &graph_out, lib_def.get()); in Encapsulate()
541 graph_out.get(), flr, lib_def.get()); in Encapsulate()
548 *library = lib_def->ToProto(); in Encapsulate()
623 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
625 GraphDefBuilder b2(GraphDefBuilder::kFailImmediately, lib_def.get()); in TEST()
629 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST()
683 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils_test.cc792 FunctionLibraryDefinition lib_def(OpRegistry::Global(), in TEST() local
795 EXPECT_FALSE(IsFunctionStateful(lib_def, *function)); in TEST()
798 EXPECT_TRUE(IsNodeStateful(lib_def, *nodeA)); in TEST()
823 EXPECT_FALSE(IsNodeStateful(lib_def, node)); in TEST()
826 EXPECT_TRUE(IsNodeStateful(lib_def, node)); in TEST()
827 EXPECT_FALSE(IsNodeStateful(lib_def, node, true)); 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/grappler/utils/
Dfunctions_test.cc832 FunctionDefLibrary lib_def; in TEST_F() local
833 *lib_def.add_function() = func; in TEST_F()
834 *lib_def.add_function() = mul_func; in TEST_F()
835 FunctionLibraryDefinition flib(OpRegistry::Global(), lib_def); in TEST_F()