Home
last modified time | relevance | path

Searched refs:entry_function (Results 1 – 25 of 80) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dsignature_with_multiple_entry_points.mlir7 // CHECK-SAME: tf.entry_function = {inputs = "input1:0,input2:0", outputs = "result:0"}
14 // CHECK-SAME: tf.entry_function = {inputs = "input2:0,input1:0", outputs = "result:0"}
18 …nsor<?xf32> {tf_saved_model.index_path = ["result"]}) attributes {tf.entry_function = {control_out…
24 …nsor<?xf32> {tf_saved_model.index_path = ["result"]}) attributes {tf.entry_function = {control_out…
Dinput_output_names_attr.mlir3 // Tests input and output names from FlatBuffer are added to `tf.entry_function` attribute.
7 // CHECK: attributes {tf.entry_function = {inputs = "input0,input1", outputs = "output0,output1"}}
8 attributes {tf.entry_function = {inputs = "input0,input1", outputs = "output0,output1"}} {
Dsignature.mlir7 // CHECK-SAME: tf.entry_function = {inputs = "serving_default_input2:0,serving_default_input1:0", o…
10 …x5xf32> {tf_saved_model.index_path = ["end_logits"]}) attributes {tf.entry_function = {control_out…
Dpruning_function_input_as_output.mlir4 func.func @main(%arg0: tensor<4xf32>) -> tensor<4xf32> attributes {tf.entry_function = {inputs = "m…
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc56 mlir::func::FuncOp entry_function = nullptr; in HandleInputOutputArraysWithModule() local
62 if (entry_function != nullptr) { in HandleInputOutputArraysWithModule()
66 entry_function = func; in HandleInputOutputArraysWithModule()
69 if (entry_function == nullptr) { in HandleInputOutputArraysWithModule()
75 entry_function->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); in HandleInputOutputArraysWithModule()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/tests/
Dinsert_main_function.mlir25 …or<1xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {inputs = "m…
29 …tensor<1xf32>, %arg1: tensor<1xf32>) -> tensor<1xf32> attributes {tf.entry_function = {inputs = "m…
34 …or<1xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {inputs = "m…
40 …tensor<1xf32>, %arg1: tensor<1xf32>) -> tensor<1xf32> attributes {tf.entry_function = {inputs = "m…
47 …tf_saved_model.index_path = ["PartitionedCall_1:0"]}) attributes {tf.entry_function = {inputs = "m…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_saved_model_initialize_variables_in_session_init.mlir8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
29 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
55 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
69 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
90 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
Dtf_saved_model_lift_variables.mlir8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
32 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
42 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
70 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
Dtf_saved_model_lift_variables_invalid_session.mlir8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
25 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
Doptimize-arg-operand-constraint.mlir7 attributes {tf.entry_function = {inputs = "input", outputs = "output_node"}} {
Dtf_saved_model_initialize_variables_in_session_init_fail.mlir7 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
Dgraph_pruning.mlir158 // "tf.entry_function" attribute defined is ignored by the pruning pass: this
175 // "tf.entry_function" attribute defined with no feed/fetch/target nodes is
178 func.func @main() attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = ""}…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf-variables.mlir7 …1x10xi64> {tf_saved_model.index_path = ["output_1"]}) attributes {tf.entry_function = {control_out…
53 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
74 …(tensor<1x10xui64> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {co…
88 …() attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = ""}, tf_saved_mod…
104 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
Dinsert_call_once_op.mlir19 …attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "hash_table_L…
35 …attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "output:0"}, …
Dpost-quantize-dynamic-range.mlir7 …CustomOp(%arg0: tensor<1x1x1x1xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
22 …CustomOp(%arg0: tensor<1x1x1x1xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
42 …CustomOp(%arg0: tensor<1x1x1x1xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
70 …32>) -> (tensor<*xf32>, tensor<*xf32>, tensor<*xf32>) attributes {tf.entry_function = {inputs = "i…
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/
Dtest.mlir10 tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "result:0"},
21 tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "result:0"},
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
Dtf_to_corert_pipeline_cpurt.mlir25 attributes { tf.entry_function = {control_outputs = "",
61 attributes { tf.entry_function = {control_outputs = "",
102 attributes { tf.entry_function = {control_outputs = "",
142 attributes { tf.entry_function = {control_outputs = "",
173 attributes { tf.entry_function = {control_outputs = "",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dexport_main_to_flib.mlir4 func.func @main() attributes {tf.entry_function = {inputs = "", outputs = ""}} {
Daliasing_arg_attr.mlir7 attributes {tf.entry_function = {inputs = "args_0,args_1", outputs = "rets_0,rets_1"}} {
Dunique_output_name.mlir4 …x1x2xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {inputs = "s…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dmetadata.mlir7 attributes {tf.entry_function = {inputs = "input", outputs = "SameNameAsOutput"}} {
Dtf_entry_function.mlir4 func.func @serving_default(%arg0: tensor<3x2xf32>) -> tensor<3x2xf32> attributes {tf.entry_function…
Dsignature_def_with_no_inputs.mlir43 …xf32> {tf_saved_model.index_path = ["start_logits"]}) attributes {tf.entry_function = {control_out…
Dunranked_tensor.mlir4 func.func @serving_default(%arg0: tensor<*xf32>) -> tensor<*xf32> attributes {tf.entry_function = {…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dunroll_batch_matmul_disabled.pbtxt80 … %[[VAL_1:.*]]: tensor<3x7xf32>) -> tensor<2x5x7xf32> attributes {tf.entry_function = {control_out…

1234