| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 45 /* static */ constexpr const char* const FunctionLibraryDefinition::kArgOp;
47 FunctionLibraryDefinition::kDeviceArgOp;
48 /* static */ constexpr const char* const FunctionLibraryDefinition::kRetOp;
50 FunctionLibraryDefinition::kDeviceRetOp;
52 FunctionLibraryDefinition::kIntsOnDeviceAttr;
53 /* static */ constexpr const char* const FunctionLibraryDefinition::kGradientOp;
54 /* static */ constexpr const char* const FunctionLibraryDefinition::kFuncAttr;
194 gnode->set_op(FunctionLibraryDefinition::kDeviceArgOp); in BuildInputArgIndex()
196 gnode->set_op(FunctionLibraryDefinition::kArgOp); in BuildInputArgIndex()
367 gnode->set_op(FunctionLibraryDefinition::kDeviceRetOp); in AddReturnNode()
[all …]
|
| D | function.h | 366 class FunctionLibraryDefinition : public OpRegistryInterface {
383 FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
384 FunctionLibraryDefinition(const OpRegistryInterface* default_registry,
386 ~FunctionLibraryDefinition() override;
388 FunctionLibraryDefinition& operator=(const FunctionLibraryDefinition&) =
446 Status AddLibrary(const FunctionLibraryDefinition& other)
507 FunctionLibraryDefinition ReachableDefinitions(const GraphDef& graph) const;
508 FunctionLibraryDefinition ReachableDefinitions(const FunctionDef& func) const;
518 const FunctionLibraryDefinition& other)
691 const FunctionLibraryDefinition* lib_def = nullptr;
[all …]
|
| D | function_test.cc | 553 (*fdef.mutable_attr())[FunctionLibraryDefinition::kIntsOnDeviceAttr].set_b( in TEST()
973 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
981 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1011 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1036 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
1058 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1072 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST()
1085 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1099 FunctionLibraryDefinition lib_def2(OpRegistry::Global(), proto); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.h | 125 const FunctionLibraryDefinition* fld);
133 const Node& node, const FunctionLibraryDefinition* fld);
142 Graph* graph, Node* node, FunctionLibraryDefinition* fld,
198 Graph* g, const FunctionLibraryDefinition* lookup_fld,
199 FunctionLibraryDefinition* fld);
203 FunctionLibraryDefinition* fld);
213 FunctionLibraryDefinition* fld);
|
| D | graph_compiler_util.cc | 84 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()
|
| D | tf2xla_util.cc | 83 const FunctionLibraryDefinition* lookup_fld, in CopyAssociatedFunctions()
84 FunctionLibraryDefinition* fld) { in CopyAssociatedFunctions()
177 const FunctionLibraryDefinition* lookup_fld, in PropagateConstIntoFuncAttr()
178 FunctionLibraryDefinition* fld) { in PropagateConstIntoFuncAttr()
222 const FunctionLibraryDefinition* lookup_fld, in PropagateConstIntoIfNode()
223 FunctionLibraryDefinition* fld) { in PropagateConstIntoIfNode()
252 const FunctionLibraryDefinition* lookup_fld, in PropagateConstIntoWhileNode()
253 FunctionLibraryDefinition* fld) { in PropagateConstIntoWhileNode()
565 const FunctionLibraryDefinition* fld) { in HasAssociatedFunction()
570 if (node_def.op() == FunctionLibraryDefinition::kGradientOp) { in HasAssociatedFunction()
[all …]
|
| D | functionalize_control_flow.cc | 71 FunctionLibraryDefinition* fld) { in GetNewFunctionName()
99 Graph* graph, FunctionLibraryDefinition* fld) { in AddFunctionDefToGraphLibrary()
134 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr,
141 FuncMap* func_map, Graph* graph, FunctionLibraryDefinition* fld, in FunctionalizeControlFlowForNodeAssociatedFunctions()
207 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr, in FunctionalizeControlFlowForFunction()
274 FunctionLibraryDefinition* library, in FunctionalizeControlFlow()
312 FunctionLibraryDefinition* library, in FunctionalizeControlFlowForGraphDef()
|
| D | resource_util_test.cc | 50 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()
|
| D | rearrange_function_argument.h | 37 Graph* g, FunctionLibraryDefinition* fld,
38 const FunctionLibraryDefinition* global_fld = nullptr);
|
| D | functionalize_cond.h | 40 Status FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library,
187 static Status Functionalize(Graph* graph, FunctionLibraryDefinition* library,
209 FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library,
275 FunctionLibraryDefinition* library_;
|
| D | functionalize_control_flow.h | 52 FunctionLibraryDefinition* library,
57 FunctionLibraryDefinition* library,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 55 explicit ClientGraph(std::unique_ptr<FunctionLibraryDefinition> flib, in ClientGraph()
65 std::unique_ptr<FunctionLibraryDefinition> flib_def;
143 std::unique_ptr<FunctionLibraryDefinition>* optimized_flib);
150 const FunctionLibraryDefinition& flib_def() const { return *flib_def_; } in flib_def()
171 std::unique_ptr<FunctionLibraryDefinition>&& flib_def,
208 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
|
| D | placer_inspection_required_ops_utils.h | 55 const FunctionLibraryDefinition* flib_def);
68 const FunctionLibraryDefinition& flib_def_;
77 Status GetFunctionDefAndAttrs(const FunctionLibraryDefinition& flib_def,
154 const FunctionLibraryDefinition& flib_def, Graph* graph);
|
| D | function_body.cc | 34 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()
|
| D | function_def_utils.h | 29 class FunctionLibraryDefinition; variable
35 const FunctionLibraryDefinition* lib_def,
43 const FunctionLibraryDefinition* lib_def,
|
| D | function.cc | 57 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp;
59 FunctionLibraryDefinition::kDeviceArgOp;
60 static constexpr const char* const kRetOp = FunctionLibraryDefinition::kRetOp;
62 FunctionLibraryDefinition::kDeviceRetOp;
64 FunctionLibraryDefinition::kGradientOp;
67 FunctionLibraryDefinition::kFuncAttr;
149 const FunctionLibraryDefinition* lib_def) in FunctionLibraryRuntimeOverlay()
180 const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
193 Status Clone(std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
200 const FunctionLibraryDefinition* lib_def_; // not owned
[all …]
|
| D | lower_function_call_op.h | 23 class FunctionLibraryDefinition; variable
32 const FunctionLibraryDefinition& flib_def,
|
| D | process_function_library_runtime.h | 68 int graph_def_version, const FunctionLibraryDefinition* lib_def,
216 const FunctionLibraryDefinition* GetFunctionLibraryDefinition() const { in GetFunctionLibraryDefinition()
266 FunctionLibraryDefinition&& lib_def, in MultiDeviceFunctionData()
282 FunctionLibraryDefinition lib_def_;
359 std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
424 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
450 const FunctionLibraryDefinition* lib_def_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.h | 24 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp;
25 static const char* const kDeviceArgOp = FunctionLibraryDefinition::kDeviceArgOp;
26 static const char* const kRetOp = FunctionLibraryDefinition::kRetOp;
27 static const char* const kDeviceRetOp = FunctionLibraryDefinition::kDeviceRetOp;
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 89 const FunctionLibraryDefinition&, int,
163 const FunctionLibraryDefinition& flib,
173 const FunctionLibraryDefinition& flib,
180 const FunctionLibraryDefinition& flib,
|
| D | functions_test.cc | 128 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 …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 38 std::unique_ptr<FunctionLibraryDefinition> flib_def,
78 State(std::shared_ptr<FunctionLibraryDefinition> flib_def, in State()
96 std::shared_ptr<FunctionLibraryDefinition> flib_def() { return flib_def_; } in flib_def()
115 std::shared_ptr<FunctionLibraryDefinition> flib_def_;
161 std::unique_ptr<FunctionLibraryDefinition>* flib_def,
185 std::unique_ptr<FunctionLibraryDefinition> flib_def,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime_test.cc | 57 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()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | add_default_attributes.cc | 28 std::unique_ptr<FunctionLibraryDefinition> flib_def( in AddDefaultAttributes()
29 new FunctionLibraryDefinition(OpRegistry::Global(), in AddDefaultAttributes()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | dump_graph.h | 52 const FunctionLibraryDefinition* flib_def = nullptr,
65 const FunctionLibraryDefinition* flib_def,
|