| /external/tensorflow/tensorflow/core/framework/ |
| function_test.cc | 972 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
974 EXPECT_EQ(lib_def.Find("XTimes16"), nullptr);
984 auto found = lib_def.Find("XTimesTwo");
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
995 EXPECT_TRUE(!lib_def.LookUpOpDef("XTimes16", &op_def).ok());
997 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def));
1003 TF_EXPECT_OK(lib_def.LookUp("XTimesTwo", &op_reg_data));
1010 // Add one function to the proto lib before constructing 'lib_def'.
1013 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
1016 TF_EXPECT_OK(lib_def.AddFunctionDef(test::function::WXPlusB()))
[all...] |
| function.h | 314 // Note: This constructor grabs `lib_def`'s lock in shared mode.
315 FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
317 const FunctionDefLibrary& lib_def);
326 // Returns nullptr if "func" is not defined in "lib_def". Otherwise,
371 // Adds the functions and gradients in 'lib_def' to this function library.
374 Status AddLibrary(const FunctionDefLibrary& lib_def) LOCKS_EXCLUDED(mu_);
729 const string& function_name, const FunctionLibraryDefinition& lib_def,
[all...] |
| function.cc | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.h | 44 // defined in "lib_def" and executes functions on the "device".
50 // "lib_def". The caller must ensure "device" and "lib_def" outlives
58 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,
|
| function.cc | 252 // wrong lib_def we just disable creation of new kernels through overlays.
309 const FunctionLibraryDefinition* lib_def,
401 const FunctionLibraryDefinition* lib_def,
404 const FunctionLibraryDefinition* lib_def,
409 const FunctionLibraryDefinition* lib_def,
426 const FunctionLibraryDefinition* lib_def,
435 base_lib_def_(lib_def),
538 const NodeDef& ndef, const FunctionLibraryDefinition* lib_def,
555 if (lib_def->Find(ndef.op()) == nullptr) {
564 if (lib_def != base_lib_def_)
700 const FunctionLibraryDefinition* lib_def = local
860 const FunctionLibraryDefinition* lib_def; local
[all...] |
| process_function_library_runtime.cc | 47 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
52 init_result_ = parent->Instantiate(function_name, lib_def, attrs, options,
60 const FunctionLibraryDefinition* lib_def,
66 lib_def_(lib_def),
85 const FunctionLibraryDefinition* lib_def,
92 lib_def_(lib_def),
480 const FunctionLibraryDefinition* lib_def,
484 auto get_func_sig = [lib_def](const string& op, const OpDef** sig) {
485 return lib_def->LookUpOpDef(op, sig);
490 FunctionDefToBodyHelper(*fdef, attrs, lib_def, get_func_sig, &tmp_fbody))
542 const FunctionLibraryDefinition* lib_def = local
[all...] |
| process_function_library_runtime.h | 32 // DeviceMgr. Caller needs to make sure that device_mgr, lib_def and parent
36 const FunctionLibraryDefinition* lib_def,
44 const FunctionLibraryDefinition* lib_def,
274 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
|
| process_function_library_runtime_test.cc | 47 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
222 std::unique_ptr<FunctionLibraryDefinition> lib_def(
228 lib_def.get(), opts, nullptr, nullptr /* cluster_flr */));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| cluster_function_library_runtime_test.cc | 60 const FunctionLibraryDefinition& lib_def,
64 return cluster_flr_->Instantiate(function_name, lib_def, attrs, options,
69 const string& function_name, const FunctionLibraryDefinition& lib_def,
74 TF_RETURN_IF_ERROR(cluster_flr_->Instantiate(function_name, lib_def, attrs,
311 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
317 TF_EXPECT_OK(InstantiateAndRun("XTimesTwoInt32", lib_def, {},
326 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
332 TF_EXPECT_OK(InstantiateAndRun("Swap", lib_def, {{"T", DT_FLOAT}},
|
| cluster_function_library_runtime.h | 38 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
|
| cluster_function_library_runtime.cc | 121 const string& function_name, const FunctionLibraryDefinition& lib_def,
138 const FunctionDef* fdef = lib_def.Find(function_name);
144 *gdef.mutable_library() = lib_def.ToProto();
|
| graph_mgr.h | 131 std::unique_ptr<FunctionLibraryDefinition> lib_def; member in struct:tensorflow::GraphMgr::Item
|
| graph_mgr.cc | 130 item->lib_def.reset(
138 TF_RETURN_IF_ERROR(graph::ValidateGraphDef(gdef, *item->lib_def));
143 item->lib_def.get(), graph_options.optimizer_options(),
191 optimization_options.flib_def = item->lib_def.get();
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass_test.cc | 493 std::unique_ptr<FunctionLibraryDefinition> lib_def(
497 std::unique_ptr<Graph> graph(new Graph(lib_def.get()));
512 device_mgr.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def.get(),
520 /*reuse_existing_functions=*/false, &graph_out, lib_def.get());
541 graph_out.get(), flr, lib_def.get());
548 *library = lib_def->ToProto();
623 std::unique_ptr<FunctionLibraryDefinition> lib_def(
625 GraphDefBuilder b2(GraphDefBuilder::kFailImmediately, lib_def.get());
629 NodeBuilder node_builder("F1", "F1", lib_def.get());
683 std::unique_ptr<FunctionLibraryDefinition> lib_def(
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| function_utils_test.cc | 792 FunctionLibraryDefinition lib_def(OpRegistry::Global(),
795 EXPECT_FALSE(IsFunctionStateful(lib_def, *function));
798 EXPECT_TRUE(IsNodeStateful(lib_def, *nodeA));
823 EXPECT_FALSE(IsNodeStateful(lib_def, node));
826 EXPECT_TRUE(IsNodeStateful(lib_def, node));
827 EXPECT_FALSE(IsNodeStateful(lib_def, node, true));
|
| vectorization_utils_test.cc | 262 FunctionLibraryDefinition lib_def(OpRegistry::Global(), lib);
264 lib_def.Find(map_defun_node.attr().at("f").func().name());
479 FunctionLibraryDefinition lib_def(OpRegistry::Global(), lib);
481 lib_def.Find(map_defun_node.attr().at("f").func().name());
[all...] |
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| functions_test.cc | 832 FunctionDefLibrary lib_def; local
833 *lib_def.add_function() = func;
834 *lib_def.add_function() = mul_func;
835 FunctionLibraryDefinition flib(OpRegistry::Global(), lib_def); local
|
