| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | rebatch.cc | 187 FunctionLibraryDefinition* flib, in RecursivelyHandleOp() argument
196 RecursivelyHandleOp(*input_node, num_workers, flib, graph)); in RecursivelyHandleOp()
203 RecursivelyHandleOp(*input_node, num_workers, flib, graph)); in RecursivelyHandleOp()
206 const FunctionDef* fdef = flib->Find(func_name); in RecursivelyHandleOp()
209 *fdef, *flib, graph->graph()->versions().producer(), &f_item)); in RecursivelyHandleOp()
217 if (flib->Find(func_def.signature().name()) == nullptr) { in RecursivelyHandleOp()
218 TF_RETURN_IF_ERROR(flib->AddFunctionDef(func_def)); in RecursivelyHandleOp()
225 TF_RETURN_IF_ERROR(MakeFunctionDef(f_item, *flib, &optimized_func)); in RecursivelyHandleOp()
228 TF_RETURN_IF_ERROR(flib->ReplaceFunction(func_name, optimized_func)); in RecursivelyHandleOp()
249 FunctionLibraryDefinition flib(OpRegistry::Global(), item.graph.library()); in OptimizeGraph() local
[all …]
|
| D | auto_shard.cc | 143 const FunctionLibraryDefinition& flib) { in ReaderOpInFunction() argument
144 const FunctionDef* func = flib.Find(node.attr().at("f").func().name()); in ReaderOpInFunction()
153 ReaderOpInFunction(func->node_def(i), flib)) { in ReaderOpInFunction()
177 FunctionLibraryDefinition* flib, in RecursivelyHandleOp() argument
189 flib, graph, nodes_to_delete)); in RecursivelyHandleOp()
203 ReaderOpInFunction(node, *flib)) { in RecursivelyHandleOp()
223 return RecursivelyHandleOp(*input_node, num_workers, index, flib, graph, in RecursivelyHandleOp()
231 FunctionLibraryDefinition flib(OpRegistry::Global(), item.graph.library()); in OptimizeGraph() local
245 TF_RETURN_IF_ERROR(RecursivelyHandleOp(sink_node, num_workers, index, &flib, in OptimizeGraph()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 244 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
249 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
326 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
331 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
381 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
382 TF_ASSERT_OK(flib.AddFunctionDef(FunctionDefHelper::Define( in TEST_F()
422 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
480 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
485 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
529 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
[all …]
|
| D | functions.h | 239 const GrapplerFunctionItem& item, const FunctionLibraryDefinition& flib,
262 const FunctionLibraryDefinition& flib,
272 const FunctionLibraryDefinition& flib,
279 const FunctionLibraryDefinition& flib,
|
| D | functions.cc | 50 Status RegisterFunctionBodyOutputs(const FunctionLibraryDefinition& flib, in RegisterFunctionBodyOutputs() argument
54 TF_RETURN_IF_ERROR(flib.LookUp(node.op(), ®istration)); in RegisterFunctionBodyOutputs()
483 const FunctionLibraryDefinition& flib, in MakeGrapplerFunctionItem() argument
513 *function_body.mutable_library() = flib.ReachableDefinitions(func).ToProto(); in MakeGrapplerFunctionItem()
518 flib.num_functions() - function_body.library().function_size(), in MakeGrapplerFunctionItem()
585 TF_RETURN_IF_ERROR(flib.LookUp(func_def_node.op(), ®istration)); in MakeGrapplerFunctionItem()
657 const FunctionLibraryDefinition& flib, in MakeGrapplerFunctionItem() argument
660 return MakeGrapplerFunctionItem(func, AttrSlice(), flib, graph_def_version, in MakeGrapplerFunctionItem()
667 const GrapplerFunctionItem& item, const FunctionLibraryDefinition& flib, in RegisterGrapplerFunctionConnectivity() argument
674 RegisterFunctionBodyOutputs(flib, func_body_node, connectivity)); in RegisterGrapplerFunctionConnectivity()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 41 const string& else_fn_name, const FunctionLibraryDefinition& flib,
104 const FunctionLibraryDefinition& flib, Graph* graph) in CondBuilder() argument
107 flib_(flib), in CondBuilder()
240 const FunctionLibraryDefinition& flib, Graph* g) { in InlineCallInGraph() argument
241 const FunctionDef* fdef = flib.Find(n->type_string()); in InlineCallInGraph()
245 FunctionDefToBodyHelper(*fdef, n->attrs(), &flib, in InlineCallInGraph()
246 [&flib](const string& op, const OpDef** sig) { in InlineCallInGraph()
247 return flib.LookUpOpDef(op, sig); in InlineCallInGraph()
293 Status RewriteIfNode(Node* n, Graph* g, const FunctionLibraryDefinition& flib) { in RewriteIfNode() argument
303 CondBuilder cb(n, then_attr->func().name(), else_attr->func().name(), flib, in RewriteIfNode()
|
| D | lower_if_while.cc | 59 FunctionLibraryDefinition* flib = options.flib_def; in Run() local
60 if (flib == nullptr) { in Run()
77 TF_RETURN_IF_ERROR(RewriteIfNode(n, g, *flib)); in Run()
79 TF_RETURN_IF_ERROR(RewriteWhileNode(n, g, *flib)); in Run()
|
| D | lower_while_op.cc | 57 Graph* graph, const FunctionLibraryDefinition& flib) { in Run() argument
59 parallel_iterations, graph, flib); in Run()
69 Graph* graph, const FunctionLibraryDefinition& flib);
155 const FunctionLibraryDefinition& flib) in LowerWhileHelper() argument
158 flib_(flib), in LowerWhileHelper()
415 const FunctionLibraryDefinition& flib) { in RewriteWhileNode() argument
432 parallel_iterations_attr->i(), g, flib)); in RewriteWhileNode()
|
| D | graph_execution_state.h | 54 explicit ClientGraph(std::unique_ptr<FunctionLibraryDefinition> flib, in ClientGraph()
57 : flib_def(std::move(flib)), in ClientGraph()
|
| D | lower_if_op.h | 25 Status RewriteIfNode(Node* n, Graph* g, const FunctionLibraryDefinition& flib);
|
| D | lower_while_op.h | 27 const FunctionLibraryDefinition& flib);
|
| D | function_threadpool_test.cc | 52 void Init(const std::vector<FunctionDef>& flib, in Init() argument
62 for (const auto& fdef : flib) *(proto.add_function()) = fdef; in Init()
|
| D | executor.h | 225 Status CreateNonCachedKernel(Device* device, FunctionLibraryRuntime* flib,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 554 FunctionLibraryDefinition flib = minimized_flib(*optimized_graph); in Optimize() local
579 if (!flib.Contains(func_name)) continue; in Optimize()
600 func, flib, trimmed_item.graph.versions().producer(), &func_item)); in Optimize()
632 if (flib.Find(func_def.signature().name()) == nullptr) { in Optimize()
633 TF_RETURN_IF_ERROR(flib.AddFunctionDef(func_def)); in Optimize()
640 TF_RETURN_IF_ERROR(MakeFunctionDef(func_item, flib, &optimized_func)); in Optimize()
643 TF_RETURN_IF_ERROR(flib.ReplaceFunction(func_name, optimized_func)); in Optimize()
648 *optimized_graph->mutable_library() = flib.ToProto(); in Optimize()
716 FunctionLibraryDefinition* flib, const DeviceSet& device_set, in OptimizeGraph() argument
743 if (flib) { in OptimizeGraph()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils_test.cc | 123 for (const auto& flib : {flib_0, flib_1}) { in TEST()
124 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("0"), make_fn_b("0"))); in TEST()
125 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("1"), make_fn_a("1"))); in TEST()
126 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("2"), make_fn_a("2"))); in TEST()
127 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("3"), make_fn_b("3"))); in TEST()
|
| D | dataset_test_base.cc | 139 const std::vector<FunctionDef>& flib, int cpu_num) { in InitFunctionLibraryRuntime() argument
153 for (const auto& fdef : flib) *(proto.add_function()) = fdef; in InitFunctionLibraryRuntime()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 59 FunctionLibraryRuntime* flib, in GraphCompiler() argument
63 flib_(flib), in GraphCompiler()
|
| D | graph_compiler.cc | 193 Status GetFunctionNameAndAttr(const FunctionLibraryRuntime& flib, in GetFunctionNameAndAttr() argument
208 if (flib.GetFunctionLibraryDefinition()->Find(node.def().op())) { in GetFunctionNameAndAttr()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test_helper.cc | 57 FunctionDefLibrary flib; in MarkForCompilation() local
58 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation()
|
| D | create_xla_launch_op_test.cc | 58 void Init(const std::vector<FunctionDef>& flib) { in Init() argument
67 for (const auto& fdef : flib) { in Init()
|
| D | encapsulate_xla_computations_pass_test.cc | 302 FunctionDefLibrary flib; in TEST() local
303 TF_ASSERT_OK(GraphToFunctionDef(*body_graph, "launch0", flib.add_function())); in TEST()
305 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
311 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST()
|
| D | mark_for_compilation_pass_test.cc | 209 FunctionDefLibrary flib; in TEST() local
210 *flib.add_function() = compilable; in TEST()
211 *flib.add_function() = uncompilable; in TEST()
212 *flib.add_function() = noinline; in TEST()
213 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
399 FunctionDefLibrary flib; in TEST() local
400 FunctionLibraryDefinition flib_def(graph->op_registry(), flib); in TEST()
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.cc | 106 FunctionDefLibrary flib; in RunMain() local
113 new FunctionLibraryDefinition((*options.graph)->op_registry(), flib); in RunMain()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 1370 const FunctionLibraryDefinition& flib, in ReachableFunctions() argument
1388 const FunctionDef* func = flib.Find(func_name); in ReachableFunctions()
1438 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctions()
1442 for (const auto& func_name : flib.ListFunctionNames()) { in ReachableFunctions()
1443 const auto& func_def = flib.Find(func_name); in ReachableFunctions()
1456 const FunctionLibraryDefinition& flib, in ReachableFunctionLibraryDefinition() argument
1458 absl::flat_hash_set<string> reachable_funcs = ReachableFunctions(flib, nodes); in ReachableFunctionLibraryDefinition()
1460 FunctionLibraryDefinition reachable_flib(flib.default_registry(), in ReachableFunctionLibraryDefinition()
1464 const FunctionDef* func = flib.Find(func_name); in ReachableFunctionLibraryDefinition()
1471 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctionLibraryDefinition()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 131 const FunctionLibraryDefinition* flib = lib.GetFunctionLibraryDefinition(); in FillOutputDevices() local
132 const FunctionDef* fdef = flib->Find(func_->name()); in FillOutputDevices()
|