| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.h | 31 FunctionLibraryDefinition* library);
32 Status FunctionalizeControlFlow(const FunctionLibraryDefinition* lookup_library,
34 FunctionLibraryDefinition* library);
37 FunctionLibraryDefinition* library);
39 const FunctionLibraryDefinition* lookup_library, GraphDef* graph_def,
40 FunctionLibraryDefinition* library);
|
| D | functionalize_control_flow.cc | 49 Status FunctionalizeControlFlow(const FunctionLibraryDefinition* lookup_library, in FunctionalizeControlFlow()
51 FunctionLibraryDefinition* library) { in FunctionalizeControlFlow()
72 FunctionLibraryDefinition* library) { in FunctionalizeControlFlow()
77 FunctionLibraryDefinition* library) { in FunctionalizeControlFlowForGraphDef()
83 const FunctionLibraryDefinition* lookup_library, GraphDef* graph_def, in FunctionalizeControlFlowForGraphDef()
84 FunctionLibraryDefinition* library) { in FunctionalizeControlFlowForGraphDef()
98 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr, in FunctionalizeControlFlowForFunction()
|
| D | tf2xla_util.h | 124 const FunctionLibraryDefinition* fld);
132 const Node& node, const FunctionLibraryDefinition* fld);
141 Graph* graph, Node* node, FunctionLibraryDefinition* fld,
197 Graph* g, const FunctionLibraryDefinition* lookup_fld,
198 FunctionLibraryDefinition* fld);
|
| 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()
223 const FunctionLibraryDefinition* lookup_fld, in PropagateConstIntoIfNode()
224 FunctionLibraryDefinition* fld) { in PropagateConstIntoIfNode()
253 const FunctionLibraryDefinition* lookup_fld, in PropagateConstIntoWhileNode()
254 FunctionLibraryDefinition* fld) { in PropagateConstIntoWhileNode()
566 const FunctionLibraryDefinition* fld) { in HasAssociatedFunction()
571 if (node_def.op() == FunctionLibraryDefinition::kGradientOp) { in HasAssociatedFunction()
[all …]
|
| D | functionalize_cond.h | 30 Status FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library);
179 static Status Functionalize(Graph* graph, FunctionLibraryDefinition* library);
200 FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library);
265 FunctionLibraryDefinition* library_;
|
| D | functionalize_while.h | 27 Status FunctionalizeWhileLoop(const FunctionLibraryDefinition* lookup_library,
28 Graph* graph, FunctionLibraryDefinition* library);
|
| D | functionalize_while.cc | 245 const FunctionLibraryDefinition* lookup_library, in AddMissingFunctionByName()
246 FunctionLibraryDefinition* library) { in AddMissingFunctionByName()
256 const FunctionLibraryDefinition* lookup_library, in AddMissingFunctionDef()
257 FunctionLibraryDefinition* library) { in AddMissingFunctionDef()
265 if (node.op() == FunctionLibraryDefinition::kGradientOp) { in AddMissingFunctionDef()
267 AttrSlice(&node.attr()).Find(FunctionLibraryDefinition::kFuncAttr); in AddMissingFunctionDef()
292 Status FunctionalizeLoop(const FunctionLibraryDefinition* lookup_library, in FunctionalizeLoop()
294 FunctionLibraryDefinition* library) { in FunctionalizeLoop()
593 Status FunctionalizeWhileLoop(const FunctionLibraryDefinition* lookup_library, in FunctionalizeWhileLoop()
595 FunctionLibraryDefinition* library) { in FunctionalizeWhileLoop()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 176 gnode->set_op(FunctionLibraryDefinition::kDeviceArgOp); in BuildInputArgIndex()
178 gnode->set_op(FunctionLibraryDefinition::kArgOp); in BuildInputArgIndex()
341 gnode->set_op(FunctionLibraryDefinition::kDeviceRetOp); in AddReturnNode()
343 gnode->set_op(FunctionLibraryDefinition::kRetOp); in AddReturnNode()
587 if (n->op() == FunctionLibraryDefinition::kArgOp || in Print()
588 n->op() == FunctionLibraryDefinition::kDeviceArgOp) { in Print()
590 } else if (n->op() == FunctionLibraryDefinition::kRetOp || in Print()
591 n->op() == FunctionLibraryDefinition::kDeviceRetOp) { in Print()
691 fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in InstantiateFunction()
692 fdef.attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b(); in InstantiateFunction()
[all …]
|
| D | function.h | 312 class FunctionLibraryDefinition : public OpRegistryInterface {
315 FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
316 FunctionLibraryDefinition(const OpRegistryInterface* default_registry,
318 ~FunctionLibraryDefinition() override;
320 FunctionLibraryDefinition& operator=(const FunctionLibraryDefinition&) =
369 Status AddLibrary(const FunctionLibraryDefinition& other) LOCKS_EXCLUDED(mu_);
439 FunctionLibraryDefinition ReachableDefinitions(const GraphDef& graph) const;
440 FunctionLibraryDefinition ReachableDefinitions(const FunctionDef& func) const;
545 const FunctionLibraryDefinition* overlay_lib = nullptr;
577 FunctionLibraryDefinition*, const DeviceSet&,
[all …]
|
| D | function_test.cc | 550 (*fdef.mutable_attr())[FunctionLibraryDefinition::kIntsOnDeviceAttr].set_b( in TEST()
972 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1013 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1048 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
1073 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST()
1087 FunctionLibraryDefinition lib_def2(OpRegistry::Global(), proto); in TEST()
1098 FunctionLibraryDefinition lib_def3(OpRegistry::Global(), proto); in TEST()
1110 FunctionLibraryDefinition lib_def4(OpRegistry::Global(), proto); in TEST()
1133 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils_test.cc | 116 FunctionLibraryDefinition flib_0(OpRegistry::Global(), fdef_base); in TEST()
119 FunctionLibraryDefinition flib_1(OpRegistry::Global(), fdef_base); in TEST()
120 FunctionLibraryDefinition flib_to_add(OpRegistry::Global(), fdef_to_add); in TEST()
150 FunctionLibraryDefinition flib_0(OpRegistry::Global(), fdef_base); in TEST()
158 FunctionLibraryDefinition flib_1(OpRegistry::Global(), fdef_base); in TEST()
159 FunctionLibraryDefinition flib_to_add(OpRegistry::Global(), fdef_to_add); in TEST()
|
| D | dataset_utils.h | 104 Status AddToFunctionLibrary(FunctionLibraryDefinition* base,
105 const FunctionLibraryDefinition& to_add);
106 Status AddToFunctionLibrary(FunctionLibraryDefinition* base,
|
| /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/common_runtime/ |
| D | function.h | 58 int graph_def_version, const FunctionLibraryDefinition* lib_def,
68 int graph_def_version, const FunctionLibraryDefinition* lib_def,
206 Status InlineFunctionBody(const FunctionLibraryDefinition& flib_def, Graph* g,
282 bool IsFunctionCall(const FunctionLibraryDefinition& lib_def, const Node& n);
288 const FunctionLibraryDefinition* const lib_def,
|
| D | process_function_library_runtime.h | 36 const FunctionLibraryDefinition* lib_def,
44 const FunctionLibraryDefinition* lib_def,
183 const FunctionLibraryDefinition& overlay_lib) in MultiDeviceFunctionData()
197 FunctionLibraryDefinition overlay_lib_;
219 std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
274 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
292 const FunctionLibraryDefinition* lib_def_;
|
| D | graph_execution_state.h | 54 explicit ClientGraph(std::unique_ptr<FunctionLibraryDefinition> flib, in ClientGraph()
64 std::unique_ptr<FunctionLibraryDefinition> flib_def;
192 std::unique_ptr<FunctionLibraryDefinition>* optimized_flib);
205 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
|
| D | function.cc | 50 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp;
52 FunctionLibraryDefinition::kDeviceArgOp;
53 static constexpr const char* const kRetOp = FunctionLibraryDefinition::kRetOp;
55 FunctionLibraryDefinition::kDeviceRetOp;
57 FunctionLibraryDefinition::kGradientOp;
60 FunctionLibraryDefinition::kFuncAttr;
170 const FunctionLibraryDefinition* overlay_lib_def) in FunctionLibraryRuntimeOverlay()
192 const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
203 Status Clone(std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
209 const FunctionLibraryDefinition* overlay_lib_def_; // not owned
[all …]
|
| D | lower_while_op.cc | 57 Graph* graph, const FunctionLibraryDefinition& flib) { in Run()
69 Graph* graph, const FunctionLibraryDefinition& flib);
133 const FunctionLibraryDefinition& flib_;
155 const FunctionLibraryDefinition& flib) in LowerWhileHelper()
377 const FunctionLibraryDefinition& lib) { in InlineCallInGraph()
415 const FunctionLibraryDefinition& flib) { in RewriteWhileNode()
|
| D | lower_if_op.cc | 41 const string& else_fn_name, const FunctionLibraryDefinition& flib,
94 const FunctionLibraryDefinition& flib_;
104 const FunctionLibraryDefinition& flib, Graph* graph) in CondBuilder()
240 const FunctionLibraryDefinition& flib, Graph* g) { in InlineCallInGraph()
293 Status RewriteIfNode(Node* n, Graph* g, const FunctionLibraryDefinition& flib) { in RewriteIfNode()
|
| D | partitioning_utils.h | 74 FunctionNameGenerator(const FunctionLibraryDefinition* flib_def, in FunctionNameGenerator()
83 const FunctionLibraryDefinition* flib_def_;
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 183 const FunctionLibraryDefinition&, int,
239 const GrapplerFunctionItem& item, const FunctionLibraryDefinition& flib,
262 const FunctionLibraryDefinition& flib,
272 const FunctionLibraryDefinition& flib,
279 const FunctionLibraryDefinition& flib,
|
| D | functions_test.cc | 244 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
326 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
381 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
480 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
529 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
594 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
633 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
660 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
701 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
741 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
[all …]
|
| /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/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 493 std::unique_ptr<FunctionLibraryDefinition> lib_def( in Encapsulate()
494 new FunctionLibraryDefinition(OpRegistry::Global(), *library)); in Encapsulate()
623 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST()
624 new FunctionLibraryDefinition(OpRegistry::Global(), library_expected)); in TEST()
683 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST()
684 new FunctionLibraryDefinition(OpRegistry::Global(), library_expected)); in TEST()
745 FunctionLibraryDefinition library(OpRegistry::Global(), {}); in TEST()
797 FunctionLibraryDefinition library(OpRegistry::Global(), {}); in TEST()
842 FunctionLibraryDefinition library(OpRegistry::Global(), {}); in TEST()
952 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 55 std::unique_ptr<FunctionLibraryDefinition> flib_def = in FromGraph()
56 MakeUnique<FunctionLibraryDefinition>(graph.flib_def()); in FromGraph()
113 FunctionLibraryDefinition* flib_def, thread::ThreadPool* pool) in Dataset()
|