Home
last modified time | relevance | path

Searched refs:tf_saved_model (Results 1 – 25 of 65) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_saved_model_ops_invalid.mlir3 module attributes {tf_saved_model.semantics} {
5 …// expected-error@+1 {{unknown tf_saved_model dialect arg attribute 'tf_saved_model.not_a_real_arg…
6 func private @f(%arg0: tensor<f32> {tf_saved_model.not_a_real_arg_attr = 1 : i32}) {
14 module attributes {tf_saved_model.semantics} {
16 // expected-error@+1 {{'tf_saved_model.bound_input' attribute should be a FlatSymbolRefAttr}}
18 %arg0: tensor<f32> {tf_saved_model.bound_input = 1 : i32}
19 ) attributes { tf_saved_model.exported_names = ["foo.some_func"] } {
27 module attributes {tf_saved_model.semantics} {
29 …// expected-error@+1 {{'tf_saved_model.bound_input' attribute must reference a valid symbol, got i…
31 %arg0: tensor<f32> {tf_saved_model.bound_input = @doesnt_exist}
[all …]
Dtf_saved_model_optimize_global_tensors.mlir7 module attributes {tf_saved_model.semantics} {
11 // CHECK: "tf_saved_model.global_tensor"() {
14 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", type = tensor<f32>, value = dense<4…
16 …f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32> {tf_…
17 attributes {tf_saved_model.exported_names = ["f"]} {
26 module attributes {tf_saved_model.semantics} {
30 // CHECK: "tf_saved_model.global_tensor"() {
33 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", type = tensor<f32>, value = dense<4…
35 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
36 attributes {tf_saved_model.exported_names = ["f"]} {
[all …]
Dtf_saved_model_ops.mlir3 module attributes {tf_saved_model.semantics} {
5 // CHECK: tf_saved_model.session_initializer
6 "tf_saved_model.session_initializer"() {
10 // CHECK: tf_saved_model.asset
11 "tf_saved_model.asset"() {
17 // CHECK: tf_saved_model.global_tensor
18 "tf_saved_model.global_tensor"() {
19 tf_saved_model.exported_names = ["some_const"],
26 // CHECK: tf_saved_model.global_tensor
27 "tf_saved_model.global_tensor"() {
[all …]
Dtf_saved_model_freeze_global_tensors.mlir3 module attributes {tf_saved_model.semantics} {
7 // CHECK-NOT: tf_saved_model.global_tensor
8 …"tf_saved_model.global_tensor"() {sym_name = "v", type = tensor<f32>, value = dense<1.0> : tensor<…
11 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
12 attributes {tf_saved_model.exported_names = ["f"]} {
22 module attributes {tf_saved_model.semantics} {
27 // CHECK: func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.index_path = [0]})
28 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.index_path = [0]})
29 attributes {tf_saved_model.exported_names = ["f"]} {
38 module attributes {tf_saved_model.semantics} {
[all …]
Dtf_saved_model_optimize_global_tensors_interprocedural.mlir6 // CHECK-LABEL: module attributes {tf_saved_model.semantics}
7 module attributes {tf_saved_model.semantics} {
12 // CHECK: "tf_saved_model.global_tensor"() {
15 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", type = tensor<f32>, value = dense<4…
17 …f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32> {tf_…
18 attributes {tf_saved_model.exported_names = ["f"]} {
36 // CHECK-LABEL: module attributes {tf_saved_model.semantics}
37 module attributes {tf_saved_model.semantics} {
42 // CHECK: "tf_saved_model.global_tensor"() {
44 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", type = tensor<f32>, value = dense<4…
[all …]
Dtf_saved_model_lift_variables.mlir3 module attributes {tf_saved_model.semantics, tf_saved_model.under_construction} {
7 …50xf32>>> {tf.resource_name = "dense/bias"}) -> (tensor<100x50xf32> {tf_saved_model.index_path = […
8 …on = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names…
16 // CHECK: "tf_saved_model.global_tensor"()
18 // CHECK: "tf_saved_model.global_tensor"()
21 …// CHECK: %arg0: tensor<!tf.resource<tensor<100x50xf32>>> {tf_saved_model.bound_input = @"dense…
22 …// CHECK: %arg1: tensor<!tf.resource<tensor<50xf32>>> {tf_saved_model.bound_input = @"dense/bia…
27 module attributes {tf_saved_model.semantics, tf_saved_model.under_construction} {
31 …50xf32>>> {tf.resource_name = "dense/bias"}) -> (tensor<100x50xf32> {tf_saved_model.index_path = […
32 …on = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names…
[all …]
Dtf_saved_model_remove_vars_in_session_initializer.mlir3 module attributes {tf_saved_model.semantics} {
9 module attributes {tf_saved_model.semantics} {
11 …// expected-error@+1 {{'tf_saved_model.session_initializer' op the initializer function does not e…
12 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
17 module attributes {tf_saved_model.semantics} {
19 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
21 …func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]}…
30 module attributes {tf_saved_model.semantics} {
35 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
36 …func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]}…
[all …]
Dtf_saved_model_deduplicate_bound_input_bindings.mlir3 module attributes {tf_saved_model.semantics, tf_saved_model.under_construction} {
5 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", type = tensor<f32>, value = dense<4…
6 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "w", type = tensor<f32>, value = dense<4…
7 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "x", type = tensor<f32>, value = dense<4…
9 // CHECK: %arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}
10 // CHECK: %arg1: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @w}
11 // CHECK: %arg2: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @x}
15 %arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v},
16 %arg1: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @w},
17 %arg2: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v},
[all …]
Dtf_saved_model_lift_variables_invalid_session.mlir5 module attributes {tf_saved_model.semantics, tf_saved_model.under_construction} {
7 …50xf32>>> {tf.resource_name = "dense/bias"}) -> (tensor<100x50xf32> {tf_saved_model.index_path = […
8 …on = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names…
21 module attributes {tf_saved_model.semantics, tf_saved_model.under_construction} {
24 func @serving_default() -> (tensor<100x50xf32> {tf_saved_model.index_path = ["dense_2"]})
25 …on = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names…
Dtf_saved_model_freeze_global_tensors_mutable_tensors.mlir3 module attributes {tf_saved_model.semantics} {
7 …"tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", type = tensor<f32>, value = dense<1…
9 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
10 attributes {tf_saved_model.exported_names = ["f"]} {
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dremove_unused_func_args_tensors.mlir4 module attributes {tf_saved_model.semantics} {
5 …"tf_saved_model.global_tensor"() {is_mutable, sym_name = "Variable", type = tensor<1x10xf32>, valu…
6 …lt(%arg0: tensor<1x10xf32> {tf_saved_model.index_path = ["x"]}, %arg1: tensor<!tf.resource<tensor<…
7 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf_saved_model.exported_…
12 …r<1x10xf32> {tf_saved_model.index_path = ["x"]}) -> (tensor<1x10xf32> {tf_saved_model.index_path =…
18 module attributes {tf_saved_model.semantics} {
19 …"tf_saved_model.global_tensor"() {is_mutable, sym_name = "Variable", type = tensor<1x10xf32>, valu…
20 …"tf_saved_model.global_tensor"() {is_mutable, sym_name = "UnusedVariable", type = tensor<1x10xf32>…
21 …lt(%arg0: tensor<1x10xf32> {tf_saved_model.index_path = ["x"]}, %arg1: tensor<!tf.resource<tensor<…
22 %arg2: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @UnusedVariable}) ->
[all …]
Dinitialize_variables.mlir4 module attributes {tf_saved_model.semantics} {
5 …"tf_saved_model.global_tensor"() {is_mutable, sym_name = "Variable", type = tensor<1x10xf32>, valu…
6 …lt(%arg0: tensor<1x10xf32> {tf_saved_model.index_path = ["x"]}, %arg1: tensor<!tf.resource<tensor<…
7 …tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {control_outputs = "", …
15 …// CHECK: func @SessionInitializerFunction() attributes {tf_saved_model.exported_names = ["Session…
21 …// CHECK: "tf_saved_model.session_initializer"() {initializers = [@SessionInitializerFunction]} : …
22 // CHECK: func @serving_default(%arg0: tensor<1x10xf32> {tf_saved_model.index_path = ["x"]}
29 module attributes {tf_saved_model.semantics} {
31 attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]} {
43 "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
[all …]
Dlegalize-tf-variables.mlir4 module attributes {tf_saved_model.semantics} {
5 …"tf_saved_model.global_tensor"() {is_mutable, sym_name = "Variable", type = tensor<1x10xf32>, valu…
6 …lt(%arg0: tensor<1x10xf32> {tf_saved_model.index_path = ["x"]}, %arg1: tensor<!tf.resource<tensor<…
7 …tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {control_outputs = "", …
15 // CHECK: func @serving_default(%arg0: tensor<1x10xf32> {tf_saved_model.index_path = ["x"]}
27 module attributes {tf_saved_model.semantics} {
29 attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]} {
37 "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
38 // CHECK: "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
40 …"tf_saved_model.global_tensor"() {is_mutable, sym_name = "Variable", type = tensor<1x10xf32>, valu…
[all …]
Dinsert_call_once_op.mlir5 module attributes {tf_saved_model.semantics} {
6 "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
9 attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"]} {
18 … @serving_default(%arg0: tensor<i64> {tf_saved_model.index_path = ["x"]}) -> (tensor<*x!tf.string>…
19 …uts = "input:0", outputs = "hash_table_Lookup/LookupTableFindV2:0"}, tf_saved_model.exported_names…
33 module attributes {tf_saved_model.semantics} {
34 …func @no_call_once(%arg0: tensor<i64> {tf_saved_model.index_path = ["x"]}) -> (tensor<i64> {tf_sav…
35 … = {control_outputs = "", inputs = "input:0", outputs = "output:0"}, tf_saved_model.exported_names…
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dinitialize_variables.cc51 tf_saved_model::GlobalTensorOp GetGlobalTensor(const SymbolTable& symbol_table, in GetGlobalTensor()
61 return symbol_table.lookup<tf_saved_model::GlobalTensorOp>(sym.getValue()); in GetGlobalTensor()
87 tf_saved_model::GlobalTensorOp GetGlobalTensorOp(mlir::Operation* op, in GetGlobalTensorOp()
119 std::set<tf_saved_model::GlobalTensorOp> tensors_to_initialize; in InitializeVariables()
149 builder.create<tf_saved_model::SessionInitializerOp>( in CreateSessionInitFunc()
164 for (auto global_tensor : module.getOps<tf_saved_model::GlobalTensorOp>()) { in runOnOperation()
Dremove_unused_func_args_tensors.cc47 std::set<mlir::tf_saved_model::GlobalTensorOp> global_tensors_to_remove; in runOnOperation()
54 symbol_table.lookup<tf_saved_model::GlobalTensorOp>( in runOnOperation()
Dinsert_call_once_op.cc36 tf_saved_model::SessionInitializerOp session_init_op = in runOnOperation()
37 tf_saved_model::GetSessionInitializerOp(module); in runOnOperation()
Dlegalize_variables.cc57 tf_saved_model::GlobalTensorOp GetGlobalTensor(const SymbolTable& symbol_table, in GetGlobalTensor()
67 return symbol_table.lookup<tf_saved_model::GlobalTensorOp>(sym.getValue()); in GetGlobalTensor()
139 for (auto global_tensor : module.getOps<tf_saved_model::GlobalTensorOp>()) { in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model_ops.td29 let name = "tf_saved_model";
39 `tf_saved_model.semantics`.
42 `tf_saved_model.exported_names = <array of strings>` which indicates which
58 `tf_saved_model.index_path` and `tf_saved_model.bound_input` in order to
70 `tf_saved_model.index_path = [...some list of strings and integers...]}`
78 `tf_saved_model.bound_input = @some_symbol_name` references a
79 `tf_saved_model.global_tensor` that is bound to that function argument
81 TODO(silvasean): Consider replacing tf_saved_model.bound_input arg attrs
85 let cppNamespace = "::mlir::tf_saved_model";
105 `tf_saved_model.bound_input` argument attr.
[all …]
Dtf_saved_model.h24 namespace tf_saved_model {
50 namespace tf_saved_model {
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dsignature_def.mlir108 … = {bad_consumers = [], min_consumer = 12 : i32, producer = 554 : i32}, tf_saved_model.semantics} {
109 …tf_saved_model.index_path = ["input2"]}, %arg1: tensor<?x384xf32> {tf_saved_model.index_path = ["i…
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_passes.cc94 mlir::tf_saved_model::CreateLiftVariablesPass(session.getValue())); in AddTFToTFLConversionPasses()
140 mlir::tf_saved_model::CreateOptimizeGlobalTensorsPass()); in AddTFToTFLConversionPasses()
174 pass_manager->addPass(mlir::tf_saved_model::CreateFreezeGlobalTensorsPass( in AddTFToTFLConversionPasses()
284 pm.addPass(mlir::tf_saved_model::CreateOptimizeGlobalTensorsPass()); in CreateTFLStandardPipeline()
300 pm.addPass(mlir::tf_saved_model::CreateFreezeGlobalTensorsPass()); in CreateTFLStandardPipeline()
314 pm.addNestedPass<mlir::tf_saved_model::SessionInitializerOp>( in CreateTFLStandardPipeline()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables_pass.cc37 if (failed(tf_saved_model::LiftVariables(module, session_))) in runOnOperation()
47 namespace tf_saved_model { namespace
Ddeduplicate_bound_input_bindings.cc26 namespace tf_saved_model { namespace
37 if (!mlir::tf_saved_model::IsExported(func)) return; in runOnFunction()
Dlift_variables_test_pass.cc125 if (failed(tf_saved_model::LiftVariables(module, session_))) in runOnOperation()
141 if (failed(tf_saved_model::LiftVariables(module, /*session=*/nullptr))) in runOnOperation()
148 namespace tf_saved_model { namespace

123