Home
last modified time | relevance | path

Searched refs:FunctionLibraryDefinition (Results 1 – 25 of 277) sorted by relevance

12345678910>>...12

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction.cc47 /* static */ constexpr const char* const FunctionLibraryDefinition::kArgOp;
49 FunctionLibraryDefinition::kDeviceArgOp;
50 /* static */ constexpr const char* const FunctionLibraryDefinition::kRetOp;
52 FunctionLibraryDefinition::kDeviceRetOp;
54 FunctionLibraryDefinition::kIntsOnDeviceAttr;
55 /* static */ constexpr const char* const FunctionLibraryDefinition::kGradientOp;
56 /* static */ constexpr const char* const FunctionLibraryDefinition::kFuncAttr;
205 gnode->set_op(FunctionLibraryDefinition::kDeviceArgOp); in BuildInputArgIndex()
207 gnode->set_op(FunctionLibraryDefinition::kArgOp); in BuildInputArgIndex()
390 gnode->set_op(FunctionLibraryDefinition::kDeviceRetOp); in AddReturnNode()
[all …]
H A Dfunction_test.cc612 (*fdef.mutable_attr())[FunctionLibraryDefinition::kIntsOnDeviceAttr].set_b( in TEST()
1040 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1048 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1059 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1078 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1103 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
1125 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1139 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1152 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1166 FunctionLibraryDefinition lib_def2(OpRegistry::Global(), proto); in TEST()
[all …]
H A Dfunction.h384 class FunctionLibraryDefinition : public OpRegistryInterface {
401 FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
402 FunctionLibraryDefinition(const OpRegistryInterface* default_registry,
404 ~FunctionLibraryDefinition() override;
406 FunctionLibraryDefinition& operator=(const FunctionLibraryDefinition&) =
465 Status AddLibrary(const FunctionLibraryDefinition& other)
529 FunctionLibraryDefinition ReachableDefinitions(const GraphDef& graph) const;
530 FunctionLibraryDefinition ReachableDefinitions(const FunctionDef& func) const;
540 const FunctionLibraryDefinition& other)
713 const FunctionLibraryDefinition* lib_def = nullptr;
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util.cc87 const FunctionLibraryDefinition* lookup_fld, in CopyAssociatedFunctions()
88 FunctionLibraryDefinition* fld) { in CopyAssociatedFunctions()
231 const FunctionLibraryDefinition* lookup_fld, FunctionLibraryDefinition* fld, in PropagateConstIntoFuncAttr()
282 const FunctionLibraryDefinition* lookup_fld, in PropagateConstIntoIfNode()
283 FunctionLibraryDefinition* fld) { in PropagateConstIntoIfNode()
313 const FunctionLibraryDefinition* lookup_fld, in FindOrInsert()
314 const FunctionLibraryDefinition* fallback_fld) { in FindOrInsert()
338 const FunctionLibraryDefinition* lookup_fld,
339 const FunctionLibraryDefinition* fallback_fld,
345 const Edge* src, const FunctionLibraryDefinition* lookup_fld, in TraverseUnmodifiedPathBackward()
[all …]
H A Dtf2xla_util.h126 const FunctionLibraryDefinition* fld);
134 const Node& node, const FunctionLibraryDefinition* fld);
143 Graph* graph, Node* node, FunctionLibraryDefinition* fld,
196 Graph* g, const FunctionLibraryDefinition* lookup_fld,
197 FunctionLibraryDefinition* fld);
201 FunctionLibraryDefinition* fld);
211 FunctionLibraryDefinition* fld);
220 const FunctionLibraryDefinition* lookup_fld);
H A Dgraph_compiler_util.cc84 FunctionLibraryDefinition::FunctionLibraryDefinition::kArgOp) in AddArgNodes()
133 FunctionLibraryDefinition::kRetOp) in AddRetvalNodes()
173 if (n->type_string() == FunctionLibraryDefinition::kArgOp) { in RewriteAndPruneGraph()
177 return errors::Aborted(FunctionLibraryDefinition::kArgOp, in RewriteAndPruneGraph()
180 } else if (n->type_string() == FunctionLibraryDefinition::kRetOp) { in RewriteAndPruneGraph()
184 return errors::Aborted(FunctionLibraryDefinition::kRetOp, in RewriteAndPruneGraph()
204 if (n->type_string() == FunctionLibraryDefinition::kArgOp) { in CollectArgNodes()
211 "Multiple ", FunctionLibraryDefinition::kArgOp, in CollectArgNodes()
222 "Expected ", FunctionLibraryDefinition::kArgOp, " node with index ", in CollectArgNodes()
269 FunctionLibraryDefinition flib_def(OpRegistry::Global(), graph_def.library()); in InitGraph()
H A Dresource_util_test.cc50 const GraphDef& graphdef, FunctionLibraryDefinition* flib_def, in AnalyzeAndVerify()
85 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
120 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
160 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
203 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
252 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
315 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST()
392 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST()
449 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST()
499 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST()
H A Dfunctionalize_control_flow.cc71 FunctionLibraryDefinition* fld) { in GetNewFunctionName()
99 Graph* graph, FunctionLibraryDefinition* fld) { in AddFunctionDefToGraphLibrary()
133 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr,
140 FuncMap* func_map, Graph* graph, FunctionLibraryDefinition* fld, in FunctionalizeControlFlowForNodeAssociatedFunctions()
206 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr, in FunctionalizeControlFlowForFunction()
273 FunctionLibraryDefinition* library, in FunctionalizeControlFlow()
311 FunctionLibraryDefinition* library, in FunctionalizeControlFlowForGraphDef()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_execution_state.h55 explicit ClientGraph(std::unique_ptr<FunctionLibraryDefinition> flib, in ClientGraph()
65 std::unique_ptr<FunctionLibraryDefinition> flib_def;
143 const FunctionLibraryDefinition* flib_def,
145 std::unique_ptr<FunctionLibraryDefinition>* optimized_flib);
155 const FunctionLibraryDefinition& flib_def() const { return *flib_def_; } in flib_def()
176 std::unique_ptr<FunctionLibraryDefinition>&& flib_def,
213 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
H A Dfunction.cc61 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp;
63 FunctionLibraryDefinition::kDeviceArgOp;
64 static constexpr const char* const kRetOp = FunctionLibraryDefinition::kRetOp;
66 FunctionLibraryDefinition::kDeviceRetOp;
68 FunctionLibraryDefinition::kGradientOp;
71 FunctionLibraryDefinition::kFuncAttr;
153 const FunctionLibraryDefinition* lib_def) in FunctionLibraryRuntimeOverlay()
184 const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
197 Status Clone(std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
204 const FunctionLibraryDefinition* lib_def_; // not owned
[all …]
H A Dprocess_function_library_runtime.h69 int graph_def_version, const FunctionLibraryDefinition* lib_def,
177 const FunctionLibraryDefinition* lib_def,
224 const FunctionLibraryDefinition* GetFunctionLibraryDefinition() const { in GetFunctionLibraryDefinition()
301 FunctionLibraryDefinition&& lib_def, in MultiDeviceFunctionData()
317 FunctionLibraryDefinition lib_def_;
380 std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
480 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
506 const FunctionLibraryDefinition* lib_def_;
H A Dplacer.h72 const FunctionLibraryDefinition* flib_def, const DeviceSet* devices,
76 const FunctionLibraryDefinition* flib_def, const DeviceSet* devices);
78 const FunctionLibraryDefinition* flib_def, const DeviceSet* devices,
98 const FunctionLibraryDefinition* const flib_def_; // Not owned.
H A Dplacer_inspection_required_ops_utils.h53 const FunctionLibraryDefinition* flib_def);
66 const FunctionLibraryDefinition& flib_def_;
75 Status GetFunctionDefAndAttrs(const FunctionLibraryDefinition& flib_def,
152 const FunctionLibraryDefinition& flib_def, Graph* graph);
H A Dfunction_body.cc34 if (n->type_string() == FunctionLibraryDefinition::kRetOp || in FunctionBody()
35 n->type_string() == FunctionLibraryDefinition::kDeviceRetOp) { in FunctionBody()
37 } else if (n->type_string() == FunctionLibraryDefinition::kArgOp || in FunctionBody()
38 n->type_string() == FunctionLibraryDefinition::kDeviceArgOp) { in FunctionBody()
H A Doptimization_registry_test.cc42 std::unique_ptr<FunctionLibraryDefinition> flib_def( in TEST()
43 new FunctionLibraryDefinition(OpRegistry::Global(), in TEST()
67 new FunctionLibraryDefinition(OpRegistry::Global(), func_def_lib)); in OptimizationPassTest()
83 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/
H A Dmlir_graph_optimization_pass_test.cc39 const FunctionLibraryDefinition& function_library));
42 const FunctionLibraryDefinition& function_library));
54 const FunctionLibraryDefinition& function_library));
69 const FunctionLibraryDefinition& function_library));
75 const FunctionLibraryDefinition& function_library) override { in Run()
111 flib_.reset(new FunctionLibraryDefinition(graph_->flib_def())); in Init()
150 std::unique_ptr<FunctionLibraryDefinition> flib_;
H A Dmlir_graph_optimization_pass.h65 const FunctionLibraryDefinition& function_library) const = 0;
69 const FunctionLibraryDefinition& function_library) = 0;
120 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def,
148 const FunctionLibraryDefinition& function_library) const = 0;
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dsplit_utils_test.cc188 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
233 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
262 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
296 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
330 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
372 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
399 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
430 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
472 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dfunctions_test.cc128 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
209 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
268 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
369 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
421 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
460 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
487 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
536 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
585 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
679 FunctionLibraryDefinition flib(OpRegistry::Global(), lib_def); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_utils.h85 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Compute()
122 OpKernelContext* ctx, std::unique_ptr<FunctionLibraryDefinition> flib_def,
198 Status AddToFunctionLibrary(FunctionLibraryDefinition* base,
199 const FunctionLibraryDefinition& to_add);
200 Status AddToFunctionLibrary(FunctionLibraryDefinition* base,
204 Status IsFunctionStateful(const FunctionLibraryDefinition& library,
208 Status IsNodeStateful(const FunctionLibraryDefinition& library,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dencapsulate_subgraphs_pass_test.cc484 std::unique_ptr<FunctionLibraryDefinition> lib_def( in Encapsulate()
485 new FunctionLibraryDefinition(OpRegistry::Global(), *library)); in Encapsulate()
616 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST()
617 new FunctionLibraryDefinition(OpRegistry::Global(), library_expected)); in TEST()
676 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST()
677 new FunctionLibraryDefinition(OpRegistry::Global(), library_expected)); in TEST()
738 FunctionLibraryDefinition library(OpRegistry::Global(), {}); in TEST()
790 FunctionLibraryDefinition library(OpRegistry::Global(), {}); in TEST()
835 FunctionLibraryDefinition library(OpRegistry::Global(), {}); in TEST()
952 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST()
[all …]
H A Dencapsulate_subgraphs_pass.cc136 Status SplitIntoSubgraphs(FunctionLibraryDefinition* library);
146 FunctionLibraryDefinition* library);
150 Status BuildOutputGraph(Graph* graph_out, FunctionLibraryDefinition* library);
187 FunctionLibraryDefinition* library);
233 Status ReplaceFunctionDef(FunctionLibraryDefinition* library);
355 FunctionLibraryDefinition* library);
545 bool reuse_existing_functions, FunctionLibraryDefinition* library) { in BuildFunctionDef()
621 FunctionLibraryDefinition* library) { in ReplaceFunctionDef()
751 Status Encapsulator::SplitIntoSubgraphs(FunctionLibraryDefinition* library) { in SplitIntoSubgraphs()
786 FunctionLibraryDefinition* library) { in BuildFunctionDefs()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunction_ops.h25 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp;
26 static const char* const kDeviceArgOp = FunctionLibraryDefinition::kDeviceArgOp;
27 static const char* const kRetOp = FunctionLibraryDefinition::kRetOp;
28 static const char* const kDeviceRetOp = FunctionLibraryDefinition::kDeviceRetOp;
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Diterator_ops.h42 std::unique_ptr<FunctionLibraryDefinition> flib_def,
83 State(std::shared_ptr<FunctionLibraryDefinition> flib_def, in State()
106 std::shared_ptr<FunctionLibraryDefinition> flib_def() { return flib_def_; } in flib_def()
127 std::shared_ptr<FunctionLibraryDefinition> flib_def_;
168 std::unique_ptr<FunctionLibraryDefinition>* flib_def,
192 std::unique_ptr<FunctionLibraryDefinition> flib_def,
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcluster_function_library_runtime_test.cc57 const FunctionLibraryDefinition& lib_def, GraphDef* g, in ConstructFunctionGraphHelper()
64 const FunctionLibraryDefinition& lib_def, in Instantiate()
74 const string& function_name, const FunctionLibraryDefinition& lib_def, in InstantiateAndRun()
123 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F()
382 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F()
397 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F()

12345678910>>...12