| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| add_default_attributes.cc | 28 std::unique_ptr<FunctionLibraryDefinition> flib_def(
33 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(output_graph_def, *flib_def, 0));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| dump_graph.h | 39 // and an optional function library 'flib_def'. Returns the file name chosen.
41 const FunctionLibraryDefinition* flib_def = nullptr);
|
| dump_graph.cc | 76 const FunctionLibraryDefinition* flib_def) {
79 if (flib_def) {
80 *graph_def.mutable_library() = flib_def->ToProto();
|
| tf2xla.cc | 268 compiler_options.flib_def = &graph->flib_def();
319 FunctionLibraryDefinition flib_def(OpRegistry::Global(), graph_def.library());
320 std::unique_ptr<Graph> g(new Graph(flib_def));
|
| xla_compiler.h | 236 const FunctionLibraryDefinition* flib_def = nullptr; member in struct:tensorflow::XlaCompiler::Options
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| optimization_registry_test.cc | 49 return options.flib_def->AddFunctionDef(test::function::WXPlusB());
67 options.flib_def = flib_def_.get();
|
| graph_execution_state.h | 54 : flib_def(std::move(flib)),
55 graph(flib_def.get()),
60 std::unique_ptr<FunctionLibraryDefinition> flib_def; member in struct:tensorflow::ClientGraph
|
| function.h | 159 void InlineFunctionBody(const FunctionLibraryDefinition& flib_def, Graph* g,
|
| graph_optimizer.cc | 97 std::unique_ptr<Graph> copy(new Graph(g->flib_def()));
|
| optimization_registry.h | 42 FunctionLibraryDefinition* flib_def = nullptr; // Not owned. member in struct:tensorflow::GraphOptimizationPassOptions
|
| direct_session.h | 152 // 'flib_def' is the function library used.
156 std::unique_ptr<FunctionLibraryDefinition> flib_def; member in struct:tensorflow::DirectSession::FunctionInfo
211 // function library 'flib_def'.
215 std::unique_ptr<FunctionLibraryDefinition>* flib_def,
|
| direct_session.cc | [all...] |
| graph_execution_state.cc | 262 optimization_options.flib_def = flib_def_.get();
425 optimization_options.flib_def = flib.get();
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.h | 49 // If specified, flib_def defines a function library that should be
51 const FunctionLibraryDefinition* flib_def = nullptr; member in struct:tensorflow::PartitionOptions
|
| graph_def_builder.h | 175 return graph_.flib_def().Find(name) != nullptr;
|
| graph_test.cc | 597 EXPECT_TRUE(graph_.flib_def().Find("XTimesTwo") != nullptr);
598 EXPECT_TRUE(graph_.flib_def().Find("XTimesFour") != nullptr);
602 EXPECT_TRUE(graph_.flib_def().Find("XTimesTwo") != nullptr);
603 EXPECT_TRUE(graph_.flib_def().Find("XTimesFour") != nullptr);
629 EXPECT_EQ(graph_.flib_def().FindGradient("XTimesTwo"), "Undefined");
633 EXPECT_EQ(graph_.flib_def().FindGradient("XTimesTwo"), "Undefined");
|
| graph_partition.cc | 1171 const FunctionLibraryDefinition* flib_def = opts.flib_def; local
[all...] |
| graph.cc | 294 Graph::Graph(const FunctionLibraryDefinition& flib_def)
295 : Graph(flib_def.default_registry()) {
297 if (flib_def.ToProto().function_size() > 0 &&
301 Status s = ops_.AddLibrary(flib_def);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| mark_for_compilation_pass_test.cc | 39 FunctionLibraryDefinition* flib_def) {
48 opt_options.flib_def = flib_def;
55 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib);
56 return MarkForCompilation(graph, &flib_def);
204 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib);
206 std::unique_ptr<Graph> graph(new Graph(&flib_def));
209 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def);
219 TF_ASSERT_OK(MarkForCompilation(&graph, &flib_def));
|
| mark_for_compilation_pass.cc | 182 const Graph& graph, FunctionLibraryDefinition* flib_def, Env* env,
188 flib_def, opts));
318 const FunctionLibraryDefinition* fld = options.flib_def;
376 *graph, options.flib_def,
571 } else if (options.flib_def->GetAttr(*n, kXlaCompileAttr, &compile_attr)
600 options.flib_def);
|
| build_xla_launch_ops_pass.cc | 138 options.flib_def);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| iterator_ops.cc | 88 std::unique_ptr<FunctionLibraryDefinition> flib_def,
92 flib_def_(std::move(flib_def)),
145 std::shared_ptr<FunctionLibraryDefinition> flib_def(
148 TF_RETURN_IF_ERROR(flib_def->AddLibrary(graph_def.library()));
161 params.function_library = flib_def;
171 lib_def_ = std::move(flib_def);
468 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr);
475 lib = CreatePrivateFLR(context, &device_mgr, &flib_def, &pflr);
478 &flib_def, &pflr, &lib));
489 [lib, &device_mgr, &flib_def, &pflr
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| dataset.cc | 144 const FunctionLibraryDefinition* flib_def = local
146 const FunctionDef* f_def = flib_def->Find(function_name);
153 const string gradient_func = flib_def->FindGradient(function_name);
164 TF_RETURN_IF_ERROR(flib_def->LookUp(node_def.op(), &op_reg_data));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.cc | 163 popts.flib_def = &graph.flib_def();
184 optimization_options.flib_def = item->lib_def.get();
|
| /external/tensorflow/tensorflow/c/ |
| c_api_function.cc | 564 return g->graph.flib_def().num_functions();
572 lib = g->graph.flib_def().ToProto();
|
