Home
last modified time | relevance | path

Searched refs:exported_names (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_saved_model_ops_invalid.mlir19 ) attributes { tf_saved_model.exported_names = ["foo.some_func"] } {
32 ) attributes { tf_saved_model.exported_names = ["foo.some_func"] } {
40 // expected-error@+1 {{'tf_saved_model.exported_names' must be on an op whose immediate parent has …
41 func @f() attributes { tf_saved_model.exported_names = ["foo.some_func"] } {
49 …// expected-error@+1 {{'tf_saved_model.exported_names' must be on a 'func' or 'tf_saved_model.glob…
51 tf_saved_model.exported_names = ["foo"]
60 // expected-error@+1 {{'tf_saved_model.exported_names' must be an array of strings}}
61 func @f() attributes { tf_saved_model.exported_names = 1 : i32} {
72 func @f() attributes { tf_saved_model.exported_names = ["foo"]} {
77 func @g() attributes { tf_saved_model.exported_names = ["foo"]} {
[all …]
Dtf_saved_model_ops.mlir19 tf_saved_model.exported_names = ["some_const"],
29 tf_saved_model.exported_names = ["some_var", "some.other.name"],
43 ) attributes { tf_saved_model.exported_names = ["some_func"] }
55 // CHECK-SAME: exported_names = ["__tf_saved_model_session_initializer"]
59 ) attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]}
74 ) attributes { tf_saved_model.exported_names = ["foo.some_func"] } {
88 ) attributes {tf_saved_model.exported_names = ["f"]} {
Dtf_saved_model_optimize_global_tensors.mlir17 attributes {tf_saved_model.exported_names = ["f"]} {
36 attributes {tf_saved_model.exported_names = ["f"]} {
53 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", tf_saved_model.exported_names = ["v…
56 attributes {tf_saved_model.exported_names = ["f"]} {
72 attributes {tf_saved_model.exported_names = ["g"]} {
90 attributes {tf_saved_model.exported_names = ["h"]} {
108 …"tf_saved_model.global_tensor"() { sym_name = "exported_unbound", tf_saved_model.exported_names = …
134 attributes {tf_saved_model.exported_names = ["f"]} {
157 attributes {tf_saved_model.exported_names = ["f"]} {
175 attributes {tf_saved_model.exported_names = ["f"]} {
[all …]
Dtf_saved_model_freeze_global_tensors.mlir12 attributes {tf_saved_model.exported_names = ["f"]} {
29 attributes {tf_saved_model.exported_names = ["f"]} {
46 attributes {tf_saved_model.exported_names = ["f"]} {
61 attributes {tf_saved_model.exported_names = ["f"]} {
83 attributes {tf_saved_model.exported_names = ["f"]} {
98 attributes {tf_saved_model.exported_names = ["f"]} {
Dtf_saved_model_remove_vars_in_session_initializer.mlir21 …func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]}…
36 …func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]}…
52 …func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]}…
73 …func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]}…
Dtf_saved_model_optimize_global_tensors_interprocedural.mlir18 attributes {tf_saved_model.exported_names = ["f"]} {
51 attributes {tf_saved_model.exported_names = ["f"]} {
57 attributes {tf_saved_model.exported_names = ["f2"]} {
82 attributes {tf_saved_model.exported_names = ["f"]} {
109 attributes {tf_saved_model.exported_names = ["f"]} {
143 attributes {tf_saved_model.exported_names = ["f"]} {
175 attributes {tf_saved_model.exported_names = ["exported_f"]} {
208 attributes {tf_saved_model.exported_names = ["exported_f"]} {
Dtf_saved_model_lift_variables.mlir8 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
32 …rol_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["f"]} {
42 …ol_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["f2"]} {
70 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
Dtf_saved_model_lift_variables_invalid_session.mlir8 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
25 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dcommon.py43 def do_test(create_module_fn, exported_names=None, show_debug_info=False): argument
62 if exported_names is None:
63 exported_names = []
88 save_model_path, ','.join(exported_names), show_debug_info)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc96 auto exported_names = GetExportedNames(init_func_op); in Verify() local
98 if (exported_names.empty()) in Verify()
102 if (exported_names.size() != 1) in Verify()
417 auto exported_names = in GetExportedNames() local
419 if (exported_names) { in GetExportedNames()
420 for (auto name : exported_names) { in GetExportedNames()
428 auto exported_names = in IsExported() local
430 return exported_names && !exported_names.empty(); in IsExported()
508 auto exported_names = GetExportedNames(init_func_op); in GetSessionInitializerExportedName() local
509 assert(exported_names.size() == 1); in GetSessionInitializerExportedName()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.h62 absl::Span<std::string> exported_names, bool add_default_attributes = true);
68 absl::Span<std::string> exported_names,
83 absl::Span<std::string> exported_names,
131 absl::Span<std::string> exported_names,
Dtf_mlir_translate.h99 absl::Span<std::string> exported_names, mlir::MLIRContext* context);
107 absl::Span<std::string> exported_names, mlir::MLIRContext* context,
117 absl::Span<std::string> exported_names, mlir::MLIRContext* context,
Dtf_mlir_translate.cc151 absl::Span<std::string> exported_names, mlir::MLIRContext* context) { in SavedModelObjectGraphToMlirImport() argument
161 auto module_or = ConvertSavedModelToMlir(&bundle, context, exported_names); in SavedModelObjectGraphToMlirImport()
171 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImport() argument
187 ConvertSavedModelV1ToMlir(bundle, exported_names, context, options); in SavedModelSignatureDefsToMlirImport()
197 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImportLite() argument
209 meta_graph_def, /*debug_info=*/{}, exported_names, context, options); in SavedModelSignatureDefsToMlirImportLite()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dremove_unused_func_args_tensors.mlir7 …f_saved_model.index_path = ["output_0"]}) attributes {tf_saved_model.exported_names = ["serving_de…
12 …f_saved_model.index_path = ["output_0"]}) attributes {tf_saved_model.exported_names = ["serving_de…
23 …f_saved_model.index_path = ["output_0"]}) attributes {tf_saved_model.exported_names = ["serving_de…
28 …f_saved_model.index_path = ["output_0"]}) attributes {tf_saved_model.exported_names = ["serving_de…
Dinsert_call_once_op.mlir9 attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]} {
19 …, outputs = "hash_table_Lookup/LookupTableFindV2:0"}, tf_saved_model.exported_names = ["serving_de…
35 …puts = "", inputs = "input:0", outputs = "output:0"}, tf_saved_model.exported_names = ["serving_de…
Dinitialize_variables.mlir7 …default_x:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_de…
15 …// CHECK: func @SessionInitializerFunction() attributes {tf_saved_model.exported_names = ["Session…
31 attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]} {
48 …default_x:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_de…
Dlegalize-tf-variables.mlir7 …default_x:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_de…
29 attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]} {
42 …default_x:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_de…
/external/tensorflow/tensorflow/compiler/mlir/
Dtf_mlir_translate_main.cc120 absl::Span<std::string> exported_names(exported_names_vector); in main() local
125 input_filename, tags, exported_names, &context); in main()
134 input_filename, tags, exported_names, &context, import_options); in main()
143 input_filename, tags, exported_names, &context, import_options); in main()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc130 absl::Span<std::string> exported_names(exported_names_in_vector); in ImportSavedModel() local
133 input_filename, tags, absl::Span<std::string>(exported_names), context); in ImportSavedModel()
140 input_filename, tags, exported_names, context, import_options); in ImportSavedModel()
/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py41 def experimental_convert_saved_model_to_mlir(saved_model_path, exported_names, argument
45 str(exported_names).encode('utf-8'), show_debug_info)
Dmlir_wrapper.cc52 const std::string &exported_names, bool show_debug_info) { in PYBIND11_MODULE() argument
56 saved_model_path, exported_names, show_debug_info, status.get()); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc151 absl::Span<std::string> exported_names(exported_names_in_vector); in ConvertSavedModelToTFLiteFlatBuffer() local
153 if (exported_names.size() != 1) { in ConvertSavedModelToTFLiteFlatBuffer()
166 exported_names, specs, &context)); in ConvertSavedModelToTFLiteFlatBuffer()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dinitialize_variables.cc106 auto exported_names = attr.dyn_cast<ArrayAttr>(); in InitializeVariables() local
107 if (!exported_names) continue; in InitializeVariables()
108 for (auto exported_name : exported_names) { in InitializeVariables()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_translate.cc168 absl::Span<std::string> exported_names(exported_names_vector); in main() local
170 if (exported_names.size() != 1) { in main()
177 tags, extra_opdefs, exported_names, in main()
Dtf_to_tfl_flatbuffer.cc229 absl::Span<std::string> exported_names, const GraphImportConfig& specs, in ImportSavedModel() argument
237 input_filename, tags, exported_names, context); in ImportSavedModel()
244 input_filename, tags, exported_names, context, options); in ImportSavedModel()

12