Home
last modified time | relevance | path

Searched refs:flib_def (Results 1 – 25 of 138) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddump_graph.cc65 const FunctionLibraryDefinition* flib_def, in DumpTextualIRToFile() argument
70 if (flib_def) { in DumpTextualIRToFile()
71 flib_def = &graph.flib_def(); in DumpTextualIRToFile()
83 flib_def ? *flib_def : graph.flib_def(), in DumpTextualIRToFile()
98 [config](const Graph& graph, const FunctionLibraryDefinition* flib_def, in UseMlirForGraphDump()
100 return DumpTextualIRToFile(config, graph, flib_def, file); in UseMlirForGraphDump()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_util_test.cc50 const GraphDef& graphdef, FunctionLibraryDefinition* flib_def, in AnalyzeAndVerify() argument
53 auto graph = absl::make_unique<Graph>(flib_def); in AnalyzeAndVerify()
59 flib_def, OptimizerOptions()); in AnalyzeAndVerify()
85 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
87 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
116 AnalyzeAndVerify(graphdef, &flib_def, expected); in TEST()
120 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
122 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
156 AnalyzeAndVerify(graphdef, &flib_def, expected); in TEST()
160 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
[all …]
Dfunctionalize_control_flow.cc120 if (graph->flib_def().LookUp(func_name, &op_reg_data).ok()) in AddFunctionDefToGraphLibrary()
331 options.flib_def); in Run()
337 TF_GRAPH_DEF_VERSION, options.flib_def, in Run()
364 string new_func_name = options.flib_def->UniqueFunctionName( in Run()
368 func.name(), new_func_name, func.attr(), options.flib_def, flr, in Run()
382 options.flib_def); in Run()
385 PruneUnreachableFunctionsFromGraph(*graph, options.flib_def)); in Run()
390 options.flib_def); in Run()
397 return FunctionalizeControlFlow(options.graph->get(), options.flib_def); in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Disolate_placer_inspection_required_ops_pass.cc41 const FunctionLibraryDefinition* flib_def = in Run() local
42 options.flib_def == nullptr ? &graph->flib_def() : options.flib_def; in Run()
43 Status status = IsolatePlacerInspectionRequiredOps(*flib_def, graph); in Run()
Dlower_function_call_op.cc46 const FunctionLibraryDefinition& flib_def, in RewriteFunctionCallNode() argument
81 fdef = flib_def.Find(func.name()); in RewriteFunctionCallNode()
86 fdef = flib_def.Find(n->type_string()); in RewriteFunctionCallNode()
95 FunctionDefToBodyHelper(*fdef, n->attrs(), &flib_def, &fbody)); in RewriteFunctionCallNode()
101 InlineFunctionBody(flib_def, g, n, fbody.get(), inline_options)); in RewriteFunctionCallNode()
Dlower_function_call_op_test.cc60 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
65 opt_options.flib_def = &flib_def; in Rewrite()
95 TF_ASSERT_OK(NodeBuilder("F", "PartitionedCall", &root.graph()->flib_def()) in TEST()
160 TF_ASSERT_OK(NodeBuilder("B", "PartitionedCall", &root.graph()->flib_def()) in TEST()
169 TF_ASSERT_OK(NodeBuilder("C", "PartitionedCall", &root.graph()->flib_def()) in TEST()
Dlower_functional_ops.cc96 FunctionLibraryDefinition* flib_def = options.flib_def; in Run() local
97 if (flib_def == nullptr) { in Run()
148 if (IsFunctionCall(*flib_def, *n) && !used_by_xla(n) && in Run()
150 TF_RETURN_IF_ERROR(RewriteFunctionCallNode(n, g, *flib_def, in Run()
Dgraph_execution_state.h58 : flib_def(std::move(flib)), in ClientGraph()
59 graph(flib_def.get()), in ClientGraph()
65 std::unique_ptr<FunctionLibraryDefinition> flib_def; member
150 const FunctionLibraryDefinition& flib_def() const { return *flib_def_; } in flib_def() function
171 std::unique_ptr<FunctionLibraryDefinition>&& flib_def,
Dplacer_inspection_required_ops_utils.cc52 : PlacerInspectionRequiredOpChecker(graph, &graph->flib_def()) {} in PlacerInspectionRequiredOpChecker()
55 const Graph* graph, const FunctionLibraryDefinition* flib_def) in PlacerInspectionRequiredOpChecker() argument
56 : graph_(*graph), flib_def_(*flib_def) { in PlacerInspectionRequiredOpChecker()
84 Status GetFunctionDefAndAttrs(const FunctionLibraryDefinition& flib_def, in GetFunctionDefAndAttrs() argument
89 *fdef = flib_def.Find(function_name); in GetFunctionDefAndAttrs()
93 "\" in function library: ", flib_def.ToProto().DebugString()); in GetFunctionDefAndAttrs()
299 const FunctionLibraryDefinition& flib_def, Graph* graph) { in IsolatePlacerInspectionRequiredOps() argument
300 PlacerInspectionRequiredOpChecker checker(graph, &flib_def); in IsolatePlacerInspectionRequiredOps()
Dplacer_inspection_required_ops_utils.h55 const FunctionLibraryDefinition* flib_def);
77 Status GetFunctionDefAndAttrs(const FunctionLibraryDefinition& flib_def,
154 const FunctionLibraryDefinition& flib_def, Graph* graph);
Dlower_functional_ops_test.cc57 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
62 opt_options.flib_def = &flib_def; in Rewrite()
162 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
267 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
Disolate_placer_inspection_required_ops_pass_test.cc42 FunctionLibraryDefinition* flib_def = nullptr) { in RunPass() argument
49 options.flib_def = flib_def; in RunPass()
152 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
153 TF_ASSERT_OK(flib_def.AddFunctionDef(func)); in TEST()
155 RunPass(original, &rewritten, &flib_def); in TEST()
Dlower_if_op_test.cc53 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
58 opt_options.flib_def = &flib_def; in Rewrite()
80 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
199 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
287 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test_helper.cc25 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in MarkForCompilation() argument
52 opt_options.flib_def = flib_def; in MarkForCompilation()
69 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation() local
70 return MarkForCompilation(graph, &flib_def, options); in MarkForCompilation()
Dforce_xla_constants_on_host_pass_test.cc41 FunctionLibraryDefinition* flib_def, in ForceXlaConstantsOnHost() argument
49 options.flib_def = flib_def; in ForceXlaConstantsOnHost()
80 FunctionLibraryDefinition flib_def(OpRegistry::Global(), library); in TEST() local
91 TF_ASSERT_OK(ForceXlaConstantsOnHost(root, &flib_def, &graph)); in TEST()
Dencapsulate_xla_computations_pass_test.cc37 const FunctionLibraryDefinition& flib_def, const string& function) { in MakeOuterGraph() argument
39 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib_def.ToProto())); in MakeOuterGraph()
50 TF_CHECK_OK(NodeDefBuilder("launch0", function, &flib_def) in MakeOuterGraph()
147 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
148 std::unique_ptr<Graph> graph(new Graph(&flib_def)); in TEST()
180 TF_CHECK_OK(EncapsulateXlaComputationsPass::Encapsulate(&graph, &flib_def)); in TEST()
205 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
206 std::unique_ptr<Graph> graph(new Graph(&flib_def)); in TEST()
253 std::unique_ptr<Graph> graph_copy(new Graph(&flib_def)); in TEST()
256 TF_ASSERT_OK(EncapsulateXlaComputationsPass::Encapsulate(&graph, &flib_def)); in TEST()
[all …]
Dresource_operation_safety_analysis_test.cc75 return ComputeIncompatibleResourceOperationPairs(*g, &g->flib_def(), {}, in ComputeIncompatiblePairs()
242 FunctionDefLibrary flib_def; in CreateFunctionDefLibWithConstFunction() local
248 *flib_def.add_function() = std::move(func); in CreateFunctionDefLibWithConstFunction()
249 return flib_def; in CreateFunctionDefLibWithConstFunction()
263 FunctionDefLibrary flib_def = in TEST() local
265 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
285 FunctionDefLibrary flib_def = in TEST() local
287 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
305 FunctionDefLibrary flib_def = in TEST() local
307 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
[all …]
/external/tensorflow/tensorflow/core/data/
Dstandalone.cc56 auto flib_def = in FromGraph() local
57 absl::make_unique<FunctionLibraryDefinition>(graph.flib_def()); in FromGraph()
60 TF_GRAPH_DEF_VERSION, flib_def.get(), OptimizerOptions{}, in FromGraph()
97 flib_def.release(), pool.release())); in FromGraph()
146 FunctionLibraryDefinition* flib_def, thread::ThreadPool* pool) in Dataset() argument
149 flib_def_(flib_def), in Dataset()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_graph_optimization_pass.cc67 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in runOnOperation() local
70 auto graph = absl::make_unique<Graph>(flib_def); in runOnOperation()
71 Status status = ConvertMlirToGraph(module_in, confs, &graph, &flib_def); in runOnOperation()
85 options.flib_def = &flib_def; in runOnOperation()
102 ConvertGraphToMlir(**options.graph, debug_info, flib_def, specs, &ctx); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dupgrade_graph.cc74 auto flib_def = library ? std::make_unique<FunctionLibraryDefinition>( in GenerateResourceSharedNameIfEmpty() local
85 TF_RETURN_IF_ERROR(flib_def->LookUpOpDef(node_def.op(), &op_def)); in GenerateResourceSharedNameIfEmpty()
100 TF_RETURN_IF_ERROR(flib_def->LookUpOpDef(node_def.op(), &op_def)); in GenerateResourceSharedNameIfEmpty()
180 Status UpgradeLegacyGraph(Graph* graph, FunctionLibraryDefinition* flib_def, in UpgradeLegacyGraph() argument
189 FunctionalizeControlFlow(graph, flib_def, node_filter, in UpgradeLegacyGraph()
/external/tensorflow/tensorflow/core/util/
Ddump_graph.cc76 const FunctionLibraryDefinition* flib_def,
186 const FunctionLibraryDefinition* flib_def, in SetGraphDumper() argument
235 const FunctionLibraryDefinition* flib_def, in DumpGraphToFile() argument
253 status = config.dumper(graph, flib_def, file.get()); in DumpGraphToFile()
265 if (flib_def) { in DumpGraphToFile()
266 *graph_def.mutable_library() = flib_def->ToProto(); in DumpGraphToFile()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Djson_translate.cc61 tensorflow::FunctionLibraryDefinition flib_def( in MlirToJSONTranslateFunction() local
64 auto graph = absl::make_unique<tensorflow::Graph>(flib_def); in MlirToJSONTranslateFunction()
65 auto status = tensorflow::ConvertMlirToGraph(module, confs, &graph, &flib_def, in MlirToJSONTranslateFunction()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h38 std::unique_ptr<FunctionLibraryDefinition> flib_def,
78 State(std::shared_ptr<FunctionLibraryDefinition> flib_def, in State() argument
82 : flib_def_(std::move(flib_def)), in State()
96 std::shared_ptr<FunctionLibraryDefinition> flib_def() { return flib_def_; } in flib_def() function
161 std::unique_ptr<FunctionLibraryDefinition>* flib_def,
185 std::unique_ptr<FunctionLibraryDefinition> flib_def,
/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_graph_optimization_pass.cc121 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run() argument
203 auto module_ref_status = ConvertGraphToMlir(**graph, debug_info, *flib_def, in Run()
287 FunctionLibraryDefinition empty_flib = empty_graph->flib_def(); in Run()
305 ConvertMlirToGraph(*module_ref, export_config, graph, flib_def, in Run()
360 ConvertGraphToMlir(**options.graph, debug_info, *options.flib_def, in Run()
383 options.flib_def), in Run()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dadd_default_attributes.cc28 std::unique_ptr<FunctionLibraryDefinition> flib_def( in AddDefaultAttributes() local
33 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(output_graph_def, *flib_def, 0)); in AddDefaultAttributes()

123456