Home
last modified time | relevance | path

Searched refs:bound_input (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_saved_model_deduplicate_bound_input_bindings.mlir4 // Test case: Remove duplicate bound_input symbols.
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},
18 %arg3: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @x},
19 %arg4: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}
Dtf_saved_model_ops.mlir39 %arg1: tensor<!tf.resource<tensor<1x64xf32>>> {tf_saved_model.bound_input = @some_constant},
40 %arg2: tensor<!tf.resource<tensor<?x64xf32>>> {tf_saved_model.bound_input = @some_variable}
57 %arg0: tensor<!tf.string> {tf_saved_model.bound_input = @asset_sym_name},
58 %arg1: tensor<!tf.resource<tensor<1x64xf32>>> {tf_saved_model.bound_input = @some_constant}
86 %arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v},
87 %arg1: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}
Dtf_saved_model_ops_invalid.mlir16 // expected-error@+1 {{'tf_saved_model.bound_input' attribute should be a FlatSymbolRefAttr}}
18 %arg0: tensor<f32> {tf_saved_model.bound_input = 1 : i32}
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}
123 …all arguments should have 'tf_saved_model.index_path', 'tf_saved_model.bound_input' or 'tf.resourc…
151 …ed_model.index_path' arg attributes should precede all 'tf_saved_model.bound_input' arg attributes…
153 %arg0: tensor<f32> {tf_saved_model.bound_input = @some_constant},
233 // expected-error@+1 {{duplicate 'tf_saved_model.bound_input' binding}}
235 %arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v},
236 %arg1: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}
[all …]
Dtf_saved_model_optimize_global_tensors.mlir16 …func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32…
35 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
55 …func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32…
88 // CHECK: func @h(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @c})
89 func @h(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @c})
133 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @c})
156 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
174 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
200 …ce<tensor<f32>>> {tf_saved_model.bound_input = @v}, %arg1: tensor<!tf.resource<tensor<f32>>> {tf_s…
225 …ce<tensor<f32>>> {tf_saved_model.bound_input = @v}, %arg1: tensor<!tf.resource<tensor<f32>>> {tf_s…
[all …]
Dtf_saved_model_lift_variables.mlir21 …// 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…
55 …// CHECK: %arg0: tensor<!tf.resource<tensor<100x50xf32>>> {tf_saved_model.bound_input = @"dense…
56 …// CHECK: %arg1: tensor<!tf.resource<tensor<50xf32>>> {tf_saved_model.bound_input = @"dense/bia…
59 …// CHECK: %arg0: tensor<!tf.resource<tensor<100x50xf32>>> {tf_saved_model.bound_input = @"dense…
60 …// CHECK: %arg1: tensor<!tf.resource<tensor<50xf32>>> {tf_saved_model.bound_input = @"dense/bia…
81 …// CHECK: %arg0: tensor<!tf.resource<tensor<100x50xf32>>> {tf_saved_model.bound_input = @"dense…
82 …// CHECK: %arg1: tensor<!tf.resource<tensor<50xf32>>> {tf_saved_model.bound_input = @"dense/bia…
Dtf_saved_model_freeze_global_tensors.mlir11 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
45 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
60 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
82 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
97 …e<tensor<f32>>> {tf_saved_model.bound_input = @"v"}, %arg2: tensor<!tf.resource<tensor<f32>>> {tf_…
Dtf_saved_model_optimize_global_tensors_interprocedural.mlir17 …func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32…
50 …func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32…
56 …func @f2(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v2}) -> (tensor<f…
81 …func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32…
107 // CHECK: func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
108 …func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32…
141 // CHECK: func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
142 …func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (tensor<f32…
174 …func @exported_f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (t…
207 …func @exported_f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) -> (t…
Dtf_saved_model_freeze_global_tensors_mutable_tensors.mlir9 func @f(%arg0: tensor<!tf.resource<tensor<f32>>> {tf_saved_model.bound_input = @v})
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dremove_unused_func_args_tensors.mlir6 … ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable}) ->
21 …h = ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable},
22 %arg2: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @UnusedVariable}) ->
28 … ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable}) -…
Dlegalize-tf-variables.mlir6 … ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable}) ->
41 … ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable}) ->
48 …h = ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable})
Dinitialize_variables.mlir6 … ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable}) ->
47 … ["x"]}, %arg1: tensor<!tf.resource<tensor<1x10xf32>>> {tf_saved_model.bound_input = @Variable}) ->
/external/tensorflow/tensorflow/python/saved_model/
Dload.py351 for bound_input, internal_capture in zip(
353 if distribute_utils.is_distributed_variable(bound_input):
355 bound_input, internal_capture)
357 concrete_function.graph.replace_capture(bound_input,
360 if resource_variable_ops.is_resource_variable(bound_input):
362 handle = bound_input.handle
370 custom_gradient.copy_handle_data(bound_input, internal_capture)
373 concrete_function.graph.capture(bound_input)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model_ops.td58 `tf_saved_model.index_path` and `tf_saved_model.bound_input` in order to
78 `tf_saved_model.bound_input = @some_symbol_name` references a
81 TODO(silvasean): Consider replacing tf_saved_model.bound_input arg attrs
105 `tf_saved_model.bound_input` argument attr.
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsaved_model_utils.cc264 for (int bound_input : saved_concrete_function.bound_inputs()) { in LoadTFConcreteFunction() local
265 auto iter = captured_objects.find(bound_input); in LoadTFConcreteFunction()
268 bound_input, in LoadTFConcreteFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3171 for (auto& bound_input : in CreateSavedModelIR() local
3173 int arg_index = bound_input_base + bound_input.index(); in CreateSavedModelIR()
3175 object_names.GetSymbolTableName(bound_input.value())); in CreateSavedModelIR()