| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 128 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
133 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
209 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
214 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
268 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
269 TF_ASSERT_OK(flib.AddFunctionDef(FunctionDefHelper::Define( in TEST_F()
309 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
369 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
374 flib, TF_GRAPH_DEF_VERSION, &item)); in TEST_F()
421 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
[all …]
|
| D | functions.cc | 216 const FunctionLibraryDefinition& flib, in MakeGrapplerFunctionItem() argument
237 FunctionDefToBodyHelper(func, func_instantiation_attr, &flib, &fbody)); in MakeGrapplerFunctionItem()
244 *function_body.mutable_library() = flib.ReachableDefinitions(func).ToProto(); in MakeGrapplerFunctionItem()
249 flib.num_functions() - function_body.library().function_size(), in MakeGrapplerFunctionItem()
291 const FunctionLibraryDefinition& flib, in MakeGrapplerFunctionItem() argument
294 return MakeGrapplerFunctionItem(func, AttrSlice(), flib, graph_def_version, in MakeGrapplerFunctionItem()
402 const FunctionLibraryDefinition& flib);
429 const GrapplerFunctionItem& item, const FunctionLibraryDefinition& flib) { in Initialize() argument
439 TF_RETURN_IF_ERROR(flib.LookUp(node.op(), ®istration)); in Initialize()
505 const FunctionLibraryDefinition& flib, in MakeFunctionDef() argument
[all …]
|
| D | functions.h | 158 const FunctionLibraryDefinition& flib,
168 const FunctionLibraryDefinition& flib,
175 const FunctionLibraryDefinition& flib,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util_test.cc | 136 FunctionDefLibrary flib; in TEST_F() local
137 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
144 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in TEST_F()
177 FunctionDefLibrary flib; in TEST_F() local
178 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
188 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
196 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in TEST_F()
257 FunctionDefLibrary flib; in TEST_F() local
258 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
265 *flib.add_function() = FunctionDefHelper::Define( in TEST_F()
[all …]
|
| D | mark_for_compilation_pass_test_helper.cc | 68 FunctionDefLibrary flib; in MarkForCompilation() local
69 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation()
|
| D | xla_kernel_creator_test.cc | 59 void Init(const std::vector<FunctionDef>& flib) { in Init() argument
68 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 | 204 FunctionDefLibrary flib; in TEST() local
205 *flib.add_function() = compilable; in TEST()
206 *flib.add_function() = uncompilable; in TEST()
207 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
244 FunctionDefLibrary flib; in TEST() local
245 *flib.add_function() = compilable; in TEST()
246 *flib.add_function() = uncompilable; in TEST()
247 *flib.add_function() = noinline; in TEST()
248 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
283 FunctionDefLibrary flib; in TEST() local
[all …]
|
| D | xla_cluster_util_test.cc | 140 FunctionDefLibrary flib; in GetNodesRelatedToRefVarsSorted() local
141 FunctionLibraryDefinition flib_def_local(OpRegistry::Global(), flib); in GetNodesRelatedToRefVarsSorted()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 240 const FunctionLibraryDefinition& flib) { in ReaderOpInFunction() argument
241 const FunctionDef* func = flib.Find(node.attr().at("f").func().name()); in ReaderOpInFunction()
250 ReaderOpInFunction(func->node_def(i), flib)) { in ReaderOpInFunction()
336 FunctionLibraryDefinition* flib, in RecursivelyHandleOp() argument
348 flib, graph, nodes_to_delete)); in RecursivelyHandleOp()
362 ReaderOpInFunction(node, *flib)) { in RecursivelyHandleOp()
386 return RecursivelyHandleOp(*input_node, num_workers, index, flib, graph, in RecursivelyHandleOp()
398 FunctionLibraryDefinition flib(OpRegistry::Global(), item.graph.library()); in OptimizeGraph() local
417 &flib, &graph, &nodes_to_delete)); in OptimizeGraph()
427 Status s = RecursivelyHandleOp(*sink_node, num_workers, index, &flib, in OptimizeGraph()
|
| D | slack_test.cc | 155 FunctionLibraryDefinition flib(OpRegistry::Global(), lib_def); in TEST() local
158 MakeGrapplerFunctionItem(*fdef, flib, /*graph_def_version=*/27, &item)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 195 FunctionDefLibrary* flib);
202 mlir::FuncOp function, FunctionDefLibrary* flib,
520 mlir::FuncOp function, FunctionDefLibrary* flib, in Convert() argument
565 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(*flib)); in Convert()
652 TF_RETURN_IF_ERROR(ConvertLibFunction(configs, tf_dialect, func, flib)); in Convert()
653 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(*flib)); in Convert()
719 FunctionDefLibrary* flib) { in ConvertLibFunction() argument
723 FunctionLibraryDefinition flib_def(&empty_registry, *flib); in ConvertLibFunction()
730 Exporter::Convert(configs, tf_dialect, function, flib, in ConvertLibFunction()
759 ConvertLibFunction(configs, tf_dialect, grad_func, flib)); in ConvertLibFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 623 FunctionLibraryDefinition flib = minimized_flib(*optimized_graph); in Optimize() local
683 if (!flib.Contains(func_name)) continue; in Optimize()
707 func, flib, trimmed_item.graph.versions().producer(), &func_item)); in Optimize()
763 if (flib.Find(func_def.signature().name()) == nullptr) { in Optimize()
764 TF_RETURN_IF_ERROR(flib.AddFunctionDef(func_def)); in Optimize()
771 TF_RETURN_IF_ERROR(MakeFunctionDef(func_item, flib, &optimized_func)); in Optimize()
774 TF_RETURN_IF_ERROR(flib.ReplaceFunction(func_name, optimized_func)); in Optimize()
779 *optimized_graph->mutable_library() = flib.ToProto(); in Optimize()
844 FunctionLibraryDefinition* flib, const DeviceSet& device_set, in OptimizeGraph() argument
873 if (flib) { in OptimizeGraph()
[all …]
|
| D | function_optimizer.cc | 440 FunctionDefLibrary PruneFunctionLibrary(const FunctionLibraryDefinition& flib, in PruneFunctionLibrary() argument
443 flib.ReachableDefinitions(optimized_graph); in PruneFunctionLibrary()
446 static_cast<int>(flib.num_functions()); in PruneFunctionLibrary()
706 const auto& flib = ctx->function_library(); in SpecializeFunction() local
712 func, func_instantiation_attr, flib, ctx->graph_version(), &item)); in SpecializeFunction()
737 TF_RETURN_IF_ERROR(MakeFunctionDef(item, flib, &specialized_func)); in SpecializeFunction()
742 if (flib.Contains(specialized_func_name)) { in SpecializeFunction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 59 FunctionLibraryRuntime* flib, in GraphCompiler() argument
63 flib_(flib), in GraphCompiler()
|
| D | graph_compiler.cc | 196 Status GetFunctionNameAndAttr(const FunctionLibraryRuntime& flib, in GetFunctionNameAndAttr() argument
211 if (flib.GetFunctionLibraryDefinition()->Find(node.def().op())) { in GetFunctionNameAndAttr()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 55 explicit ClientGraph(std::unique_ptr<FunctionLibraryDefinition> flib, in ClientGraph()
58 : flib_def(std::move(flib)), in ClientGraph()
|
| 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 | 246 Status CreateNonCachedKernel(Device* device, FunctionLibraryRuntime* flib,
|
| D | constant_folding_test.cc | 337 FunctionDefLibrary flib; in TEST_F() local
338 *flib.add_function() = test::function::XTimesTwo(); in TEST_F()
340 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST_F()
345 TF_EXPECT_OK(s.graph()->AddFunctionLibrary(flib)); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.cc | 100 FunctionDefLibrary flib; in Run() local
107 new FunctionLibraryDefinition((*options.graph)->op_registry(), flib)); in Run()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 136 const FunctionLibraryDefinition* flib = lib.GetFunctionLibraryDefinition(); in FillOutputDevices() local
137 const FunctionDef* fdef = flib->Find(func_->name()); in FillOutputDevices()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 1568 const FunctionLibraryDefinition& flib, in ReachableFunctions() argument
1585 const FunctionDef* func = flib.Find(func_name); in ReachableFunctions()
1596 for (const auto& func_name : flib.ListFunctionNames()) { in ReachableFunctions()
1597 const auto& func_def = flib.Find(func_name); in ReachableFunctions()
1651 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctions()
1659 const FunctionLibraryDefinition& flib, in ReachableFunctionLibraryDefinition() argument
1661 std::set<string> reachable_funcs = ReachableFunctions(flib, nodes); in ReachableFunctionLibraryDefinition()
1663 FunctionLibraryDefinition reachable_flib(flib.default_registry(), in ReachableFunctionLibraryDefinition()
1669 Status added = reachable_flib.CopyFunctionDefFrom(func_name, flib); in ReachableFunctionLibraryDefinition()
1672 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctionLibraryDefinition()
|
| D | op_kernel.cc | 213 const NodeDef* node_def, const OpDef* op_def, FunctionLibraryRuntime* flib, in OpKernelConstruction() argument
224 flib_(flib), in OpKernelConstruction()
1544 Allocator* allocator, FunctionLibraryRuntime* flib, in CreateOpKernel() argument
1547 return CreateOpKernel(std::move(device_type), device, allocator, flib, in CreateOpKernel()
1553 Allocator* allocator, FunctionLibraryRuntime* flib, in CreateOpKernel() argument
1609 &node_def, op_def, flib, resource_mgr, inputs, in CreateOpKernel()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils_test.cc | 192 for (const auto& flib : {flib_0, flib_1}) { in TEST()
193 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("0"), make_fn_b("0"))); in TEST()
194 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("1"), make_fn_a("1"))); in TEST()
195 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("2"), make_fn_a("2"))); in TEST()
196 EXPECT_TRUE(FunctionDefsEqual(*flib.Find("3"), make_fn_b("3"))); in TEST()
|