Home
last modified time | relevance | path

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

123

/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.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.func @f() attributes { tf_saved_model.exported_names = 1 : i32} {
72 func.func @f() attributes { tf_saved_model.exported_names = ["foo"]} {
77 func.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_initialize_variables_in_session_init.mlir8 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
15 // CHECK-SAME: {tf_saved_model.exported_names = ["SessionInitializerFunction"]}
29 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
35 func.func @Init() attributes {tf_saved_model.exported_names = ["Init"]} {
40 // CHECK-SAME: {tf_saved_model.exported_names = ["Init"]}
55 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
69 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
76 // CHECK-SAME: {tf_saved_model.exported_names = ["SessionInitializerFunction"]}
90 …utputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_de…
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.mlir13 attributes {tf_saved_model.exported_names = ["f"]} {
30 attributes {tf_saved_model.exported_names = ["f"]} {
47 attributes {tf_saved_model.exported_names = ["f"]} {
64 attributes {tf_saved_model.exported_names = ["f"]} {
86 attributes {tf_saved_model.exported_names = ["g"]} {
108 attributes {tf_saved_model.exported_names = ["f"]} {
132 attributes {tf_saved_model.exported_names = ["f"]} {
148 attributes {tf_saved_model.exported_names = ["f"]} {
Dtf_saved_model_remove_vars_in_session_initializer.mlir21 …func.func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializ…
36 …func.func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializ…
52 …func.func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializ…
73 …func.func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializ…
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_freeze_assets.mlir11 attributes {tf_saved_model.exported_names = ["f"]} {
30 attributes {tf_saved_model.exported_names = ["f"]} {
48 attributes {tf_saved_model.exported_names = ["f"]} {
67 attributes {tf_saved_model.exported_names = ["f"]} {
/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py50 def experimental_convert_saved_model_to_mlir(saved_model_path, exported_names, argument
54 str(exported_names).encode('utf-8'), show_debug_info)
58 exported_names, tags, argument
63 str(exported_names).encode('utf-8'),
68 exported_names, tags, argument
73 str(exported_names).encode('utf-8'),
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dcommon.py39 def do_test(create_module_fn, exported_names=None, show_debug_info=False): argument
58 if exported_names is None:
59 exported_names = []
84 save_model_path, ','.join(exported_names), show_debug_info)
Dcommon_v1.py101 exported_names = ''
105 save_model_path, exported_names,
116 save_model_path, exported_names,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc98 auto exported_names = GetExportedNames(init_func_op); in verify() local
100 if (exported_names.empty()) in verify()
104 if (exported_names.size() != 1) in verify()
419 auto exported_names = in GetExportedNames() local
421 if (exported_names) { in GetExportedNames()
422 for (auto name : exported_names) { in GetExportedNames()
430 auto exported_names = in IsExported() local
432 return exported_names && !exported_names.empty(); in IsExported()
510 auto exported_names = GetExportedNames(init_func_op); in GetSessionInitializerExportedName() local
511 assert(exported_names.size() == 1); in GetSessionInitializerExportedName()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dsignature_with_multiple_entry_points.mlir8 // CHECK-SAME: tf_saved_model.exported_names = ["add"]
15 // CHECK-SAME: tf_saved_model.exported_names = ["sub"]
18 …"", inputs = "input1:0,input2:0", outputs = "result:0"}, tf_saved_model.exported_names = ["add"]} {
24 …"", inputs = "input2:0,input1:0", outputs = "result:0"}, tf_saved_model.exported_names = ["sub"]} {
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.h70 absl::Span<std::string> exported_names,
78 absl::Span<std::string> exported_names,
95 std::optional<absl::Span<const std::string>> exported_names,
148 std::optional<absl::Span<const std::string>> exported_names,
Dtf_mlir_translate.cc155 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelObjectGraphToMlirImport() argument
167 &bundle, context, exported_names, /*add_default_attributes=*/true, in SavedModelObjectGraphToMlirImport()
179 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImport() argument
204 auto module_or = ConvertSavedModelV1ToMlir(*bundle_ptr, exported_names, in SavedModelSignatureDefsToMlirImport()
216 absl::Span<std::string> exported_names, mlir::MLIRContext* context, in SavedModelSignatureDefsToMlirImportLite() argument
228 if (!exported_names.empty()) optional_exported_names = exported_names; in SavedModelSignatureDefsToMlirImportLite()
Dtf_mlir_translate.h108 absl::Span<std::string> exported_names, mlir::MLIRContext* context,
118 absl::Span<std::string> exported_names, mlir::MLIRContext* context,
131 absl::Span<std::string> exported_names, mlir::MLIRContext* context,
/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/quantization/tensorflow/tests/
Dinsert_main_function.mlir20 …func.func @NoOp() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializ…
23 // CHECK: func @NoOp() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initi…
25 … = "mul1_y:0,mul1_x:0", outputs = "PartitionedCall:0"}, tf_saved_model.exported_names = ["mul1"]} {
34 … "mul2_y:0,mul2_x:0", outputs = "PartitionedCall_1:0"}, tf_saved_model.exported_names = ["mul2"]} {
47 …0", outputs = "PartitionedCall:0,PartitionedCall_1:0"}, tf_saved_model.exported_names = ["main"]} {
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc234 std::vector<string> exported_names = in ExperimentalConvertSavedModelToMlir() local
238 &bundle, &context, absl::Span<std::string>(exported_names)); in ExperimentalConvertSavedModelToMlir()
254 std::vector<string> exported_names = in ExperimentalConvertSavedModelV1ToMlirLite() local
261 saved_model_path, tag_set, absl::Span<std::string>(exported_names), in ExperimentalConvertSavedModelV1ToMlirLite()
289 std::vector<string> exported_names = in ExperimentalConvertSavedModelV1ToMlir() local
295 ConvertSavedModelV1ToMlir(bundle, absl::Span<std::string>(exported_names), in ExperimentalConvertSavedModelV1ToMlir()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf-variables.mlir7 …default_x:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_de…
34 attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]} {
53 …default_x:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_de…
74 … inputs = "", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_de…
88 … = {control_outputs = "", inputs = "", outputs = ""}, tf_saved_model.exported_names = ["serving_de…
104 … inputs = "", outputs = "StatefulPartitionedCall:0"}, 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…
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/
Dquantize_model.cc117 std::vector<std::string> exported_names = in QuantizeQatModel() local
142 absl::MakeSpan(exported_names), in QuantizeQatModel()
212 std::vector<std::string> exported_names = in QuantizePtqModelPreCalibration() local
230 absl::MakeSpan(exported_names), in QuantizePtqModelPreCalibration()
276 std::vector<std::string> exported_names = in QuantizePtqModelPostCalibration() local
301 absl::MakeSpan(exported_names), in QuantizePtqModelPostCalibration()
359 std::vector<std::string> exported_names = in QuantizePtqDynamicRange() local
384 absl::MakeSpan(exported_names), in QuantizePtqDynamicRange()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc133 absl::Span<std::string> exported_names(exported_names_in_vector); in ImportSavedModel() local
136 input_filename, tags, absl::Span<std::string>(exported_names), context); in ImportSavedModel()
143 input_filename, tags, exported_names, context, import_options); in ImportSavedModel()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/
Dhoist_invariant_ops.mlir14 attributes {tf_saved_model.exported_names = ["test_hoist_varhandleop"]} {
43 attributes {tf_saved_model.exported_names = ["test_hoist_hash_table"]} {
72 attributes {tf_saved_model.exported_names = ["test_hoist_const"]} {
85 attributes {tf_saved_model.exported_names = ["test_hoist_const_return"]} {
109 attributes {tf_saved_model.exported_names = ["test_hoist_var_read_write"]} {
141 attributes {tf_saved_model.exported_names = ["test_not_hoist_stateful_call"]} {
151 attributes {tf_saved_model.exported_names = ["test_not_hoist_if"]} {
170 func.func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initialize…
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/
Dtest.mlir11 tf_saved_model.exported_names = ["serving_default"]
22 tf_saved_model.exported_names = ["predict"]

123