Searched refs:flib_def (Results 1 - 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dadd_default_attributes.cc28 std::unique_ptr<FunctionLibraryDefinition> flib_def(
33 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(output_graph_def, *flib_def, 0));
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Ddump_graph.h39 // and an optional function library 'flib_def'. Returns the file name chosen.
41 const FunctionLibraryDefinition* flib_def = nullptr);
H A Ddump_graph.cc76 const FunctionLibraryDefinition* flib_def) {
79 if (flib_def) {
80 *graph_def.mutable_library() = flib_def->ToProto();
75 DumpGraphToFile(const string& name, Graph const& graph, const FunctionLibraryDefinition* flib_def) argument
H A Dtf2xla.cc268 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));
H A Dxla_compiler.h236 const FunctionLibraryDefinition* flib_def = nullptr; member in struct:tensorflow::XlaCompiler::Options
/external/tensorflow/tensorflow/core/common_runtime/
H A Doptimization_registry_test.cc49 return options.flib_def->AddFunctionDef(test::function::WXPlusB());
67 options.flib_def = flib_def_.get();
H A Dgraph_execution_state.h54 : flib_def(std::move(flib)),
55 graph(flib_def.get()),
60 std::unique_ptr<FunctionLibraryDefinition> flib_def; member in struct:tensorflow::ClientGraph
H A Dfunction.h159 void InlineFunctionBody(const FunctionLibraryDefinition& flib_def, Graph* g,
H A Dgraph_optimizer.cc97 std::unique_ptr<Graph> copy(new Graph(g->flib_def()));
H A Doptimization_registry.h42 FunctionLibraryDefinition* flib_def = nullptr; // Not owned. member in struct:tensorflow::GraphOptimizationPassOptions
H A Ddirect_session.h152 // '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,
H A Ddirect_session.cc1151 TF_RETURN_IF_ERROR(CreateGraphs(options, &graphs, &func_info->flib_def,
1184 func_info->flib_def.get(), optimizer_opts));
1308 std::unique_ptr<FunctionLibraryDefinition>* flib_def,
1391 popts.flib_def = &client_graph->graph.flib_def();
1419 new Graph(client_graph->flib_def.get()));
1431 optimization_options.flib_def = client_graph->flib_def.get();
1453 *flib_def = std::move(client_graph->flib_def);
1305 CreateGraphs( const BuildGraphOptions& subgraph_options, std::unordered_map<string, std::unique_ptr<Graph>>* outputs, std::unique_ptr<FunctionLibraryDefinition>* flib_def, RunStateArgs* run_state_args, DataTypeVector* input_types, DataTypeVector* output_types) argument
[all...]
H A Dgraph_execution_state.cc262 optimization_options.flib_def = flib_def_.get();
425 optimization_options.flib_def = flib.get();
/external/tensorflow/tensorflow/core/graph/
H A Dgraph_partition.h49 // If specified, flib_def defines a function library that should be
51 const FunctionLibraryDefinition* flib_def = nullptr; member in struct:tensorflow::PartitionOptions
H A Dgraph_def_builder.h175 return graph_.flib_def().Find(name) != nullptr;
H A Dgraph_test.cc597 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");
H A Dgraph_partition.cc1171 const FunctionLibraryDefinition* flib_def = opts.flib_def; local
1172 if (flib_def == nullptr) {
1173 flib_def = &g->flib_def();
1180 *gdef->mutable_library() = flib_def->ToProto();
H A Dgraph.cc294 Graph::Graph(const FunctionLibraryDefinition& flib_def) argument
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/
H A Dmark_for_compilation_pass_test.cc39 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));
38 MarkForCompilation(std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def) argument
H A Dmark_for_compilation_pass.cc182 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);
181 FindCompilationCandidates( const Graph& graph, FunctionLibraryDefinition* flib_def, Env* env, const std::function<bool(const Node*, const DeviceType&)>& is_compilable_fn, OrderedNodeSet* candidates) argument
H A Dbuild_xla_launch_ops_pass.cc138 options.flib_def);
/external/tensorflow/tensorflow/core/kernels/data/
H A Diterator_ops.cc88 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,
84 IteratorResource(const DataTypeVector& output_dtypes, const std::vector<PartialTensorShape>& output_shapes, const int , std::unique_ptr<DeviceMgr> device_mgr, std::unique_ptr<FunctionLibraryDefinition> flib_def, std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, FunctionLibraryRuntime* lib) argument
541 CreatePrivateFLR( OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, std::unique_ptr<FunctionLibraryDefinition>* flib_def, std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr) argument
[all...]
/external/tensorflow/tensorflow/core/framework/
H A Ddataset.cc144 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/
H A Dgraph_mgr.cc163 popts.flib_def = &graph.flib_def();
184 optimization_options.flib_def = item->lib_def.get();
/external/tensorflow/tensorflow/c/
H A Dc_api_function.cc564 return g->graph.flib_def().num_functions();
572 lib = g->graph.flib_def().ToProto();

Completed in 1805 milliseconds

12